try adding wait before assert