get unit tests to pass