Modified to pass the test