Remove wait argument

Removed unused wait argument