Fix a wrong import