diff --git a/src/Avalonia.Input/TextInput/InputMethodManager.cs b/src/Avalonia.Input/TextInput/InputMethodManager.cs
index e2d6a1fd9f..12305dfe33 100644
--- a/src/Avalonia.Input/TextInput/InputMethodManager.cs
+++ b/src/Avalonia.Input/TextInput/InputMethodManager.cs
@@ -38,9 +38,9 @@ namespace Avalonia.Input.TextInput
 
                     _im?.Reset();
                     
-                    if (_focusedElement is AvaloniaObject target)
+                    if (_focusedElement is StyledElement target)
                     {
-                        _im?.SetOptions(TextInputOptions.FromAvaloniaObject(target));
+                        _im?.SetOptions(TextInputOptions.FromStyledElement(target));
                     }
 
                     _transformTracker.SetVisual(_client?.TextViewVisual);