Fixed error where the class to test was mocked