BUG: fixed import problem in test suite