Fixing bug that was caused by fixing another bug

I'm dumb. I didn't checked result. Thought it was too easy.