[Fix] Fixed bad return type