Fix incorrect return