[CHORE] Fix bug