Fix logic errors. D'oh!