Fixed error with some importing of tests.