Fixing strange regression by @antineutron

Fixing your regression, @antineutron!