fix import for new repo