[TODO] fix minor error