you should be able to solve these tests by changing this method