repaired unit tests