Fix class location