fixed import in tests