Testing: Add some wait