@@ -340,26 +340,26 @@ Rectangle {
340340 initSession();
341341 setApplicationState(appRunning);
342342 tryCompare(stateGroup, "state", "surface");
343− waitUntilTransitionsEnd();
343+ waitUntilTransitionsEnd(stateGroup);
344344
345345 setApplicationState(appSuspended);
346346
347347 cleanupSession();
348348
349349 tryCompare(stateGroup, "state", "screenshot");
350− waitUntilTransitionsEnd();
350+ waitUntilTransitionsEnd(stateGroup);
351351 tryCompare(fakeApplication, "session", null);
352352
353353 // and restart it
354354 setApplicationState(appStarting);
355355
356− waitUntilTransitionsEnd();
356+ waitUntilTransitionsEnd(stateGroup);
357357 verify(stateGroup.state === "screenshot");
358358 verify(fakeSession === null);
359359
360360 setApplicationState(appRunning);
361361
362− waitUntilTransitionsEnd();
362+ waitUntilTransitionsEnd(stateGroup);
363363 verify(stateGroup.state === "screenshot");
364364
365365 initSession();