[fix] Correctly add loading classes