Fixed the previous commit (a test failed)