Fix regression

It's crap but get's the job done