Use return instead of die