fix dumb if else bug