Fixed done handling for better output in failed tests.
Merge pull request #4 from wbyoung/fixed-done

Fixed done handling for better output in failed tests.