diff --git a/geant_service_orchestrator/workflows/device/create_device.py b/geant_service_orchestrator/workflows/device/create_device.py
index d98943eb1b7d5a7970e5944381e1fad60a159f66..5be2e0e47011656048516e294ca37e76883b6fba 100644
--- a/geant_service_orchestrator/workflows/device/create_device.py
+++ b/geant_service_orchestrator/workflows/device/create_device.py
@@ -117,7 +117,7 @@ def provision_device_dry(
 
 
 @step("Confirm provisioning")
-def confirmation_form() -> FormGenerator:
+def confirmation_form() -> State:
     class ConfirmActionForm(FormPage):
         class Config:
             title = "Confirm your action"