fixed import errors on test code