Update failing test for proper mocking and expectations