Removing unnecessary Mock.mock()