Fix wrong import.