Fixed class annotation issue