Fix, should compare with 2 instead