Fix logic derp