Very minor code change to make the test green