[fix] Small correction