Got the test to pass