Updated tests to import from the mock library.