diff --git a/PHI-CL/control.lua b/PHI-CL/control.lua
index a3a067e..8164adc 100644
--- a/PHI-CL/control.lua
+++ b/PHI-CL/control.lua
@@ -87,7 +87,7 @@ if settings.startup['PHI-CT'].value then
     end
 
     function gui_update(player, inserter)
-        if not (inserter.allow_custom_vectors or inserter.supports_direction) then
+        if not (inserter.supports_direction or ((inserter.prototype and inserter.prototype.allow_custom_vectors) or (inserter.ghost_prototype and inserter.ghost_prototype.allow_custom_vectors))) then
             return
         end