Fixed the Unit tests locally