fixed import in test