[fix] small refactor