Fixed import error temporarily

Will add in the class tomorrow, forgot to stage it :S