Fixed an incorrect package import