Fix wrong import statements