Fixed import error in test module.