minor fix on wait functionality
0.2.2