Fix error in import