Updated module paths in tests.
small test fix