[fix] Small fix