diff --git a/plugins/UM3NetworkPrinting/src/Cloud/CloudOutputController.py b/plugins/UM3NetworkPrinting/src/Cloud/CloudOutputController.py
index c139be0c38..af98b55587 100644
--- a/plugins/UM3NetworkPrinting/src/Cloud/CloudOutputController.py
+++ b/plugins/UM3NetworkPrinting/src/Cloud/CloudOutputController.py
@@ -1,11 +1,11 @@
 # Copyright (c) 2018 Ultimaker B.V.
 # Cura is released under the terms of the LGPLv3 or higher.
-from UM.OutputDevice.OutputDevice import OutputDevice
 from cura.PrinterOutput.PrinterOutputController import PrinterOutputController
+from .CloudOutputDevice import CloudOutputDevice
 
 
 class CloudOutputController(PrinterOutputController):
-    def __init__(self, output_device: OutputDevice) -> None:
+    def __init__(self, output_device: CloudOutputDevice) -> None:
         super().__init__(output_device)
         
         # The cloud connection only supports fetching the printer and queue status and adding a job to the queue.