fixed implementation to mock expected output