Question: are there different size thermocouples?
Answer: You want to get one that is about the same length as the one you have. They are sold in in 6" increments for the shorter lengths, 18", 24", 30" and so on. For a water heater, 18" and 24" are common, although longer ones are possible. Remove the old thermocouple and measure it to see what length you need.