Now it passes the tests!