Fix class referencing to proper location