Fix intermitted test failure
Fix the comment
Merge pull request #198 from chinmay-gupte/fix_intermittent_test_failure

Fix intermittent test failure