fix to integer return issue