[TASK] Fixed wrong return datatype