Fixed mocking methods, added comment