Fix for the test that was not being run before.