clean up tests to make them pass