diff --git a/controllers/hetznerbaremetalhost_controller_test.go b/controllers/hetznerbaremetalhost_controller_test.go
index 8a3f9c2e4..fc244c75f 100644
--- a/controllers/hetznerbaremetalhost_controller_test.go
+++ b/controllers/hetznerbaremetalhost_controller_test.go
@@ -341,7 +341,7 @@ var _ = Describe("HetznerBareMetalHostReconciler", func() {
 						testEnv.GetLogger().Info("reaches the state image installing. Get failed", "err", err)
 						return false
 					}
-					if host.Spec.Status.ProvisioningState == infrav1.StateImageInstalling {
+					if host.Spec.Status.ProvisioningState == infrav1.StateImageInstalling || host.Spec.Status.ProvisioningState == infrav1.StateProvisioned {
 						return true
 					}
 					testEnv.GetLogger().Info("reaches the state image installing. State",