get unit tests working