Update import test to reflect new modules