[fix] add class