fixes unit tests