We should rewrite this to actually wait for something instead of blindly sleeping,
but for now let's try to make test pass by waiting more.