get unit tests passing