Fix overeager replace of return with unit.