update todo
fix import