fixed refering mocks in real code and modified test