quick fix for wait