Wait for the end (@Rahul)