class annotation fix