Fixed done handling for better output in failed tests.

A small change to make test output better.