Fix typo and return early from stage