Fix the wrong import