fixed import error in test suite