Added missing imports to test