Simplified the wait call.