Fixed class import error.