Simplified helper

I was repeating the logic twice. Duh!