Modified to pass the test.