Fix class call