Fixing all unit tests with repository