Fixed test import error caused by rename in previous commit