fix test imports from name change