Small fix on return format