xtea: Make the count parameter match the documentation