[TASK] Fix wrong data type