callback must be called at the end of test
Merge pull request #1 from amitu/test_fix

callback must be called at the end of test