fixed unit tests that failed