2 small changes which lead to passing tests!