Fix the logic, confuse myself.

I'll have to rewrite this, its super confusing now.