fixed import

To get tests working again in Travis