diff --git a/ios/Classes/FLPencilKit.swift b/ios/Classes/FLPencilKit.swift
index b98fc80..00ca976 100644
--- a/ios/Classes/FLPencilKit.swift
+++ b/ios/Classes/FLPencilKit.swift
@@ -330,6 +330,7 @@ private class PencilKitView: UIView {
     default:
       break
     }
+    toolPicker?.selectedTool = canvasView.tool
   }
 
   func save(url: URL, withBase64Data: Bool) throws -> String? {