Fixed error with running the tests.