Fix class directory