Fix incorrect return value