diff --git a/services/serviceenablement/wait/wait_test.go b/services/serviceenablement/wait/wait_test.go
index 35dd7f882..2c2a8a0cd 100644
--- a/services/serviceenablement/wait/wait_test.go
+++ b/services/serviceenablement/wait/wait_test.go
@@ -97,7 +97,7 @@ func TestEnableServiceWaitHandler(t *testing.T) {
 
 			handler := EnableServiceWaitHandler(ctx, apiClient, "eu01", "projectId", serviceId)
 
-			gotRes, err := handler.SetTimeout(10 * time.Millisecond).WaitWithContext(ctx)
+			gotRes, err := handler.SetTimeout(10 * time.Millisecond).SetSleepBeforeWait(10 * time.Millisecond).WaitWithContext(ctx)
 
 			if err != nil {
 				if !tt.wantErr {
@@ -179,7 +179,7 @@ func TestDisableServiceWaitHandler(t *testing.T) {
 
 			handler := DisableServiceWaitHandler(ctx, apiClient, "eu01", "projectId", serviceId)
 
-			gotRes, err := handler.SetTimeout(10 * time.Millisecond).WaitWithContext(ctx)
+			gotRes, err := handler.SetTimeout(10 * time.Millisecond).SetSleepBeforeWait(10 * time.Millisecond).WaitWithContext(ctx)
 
 			if err != nil {
 				if !tt.wantErr {