Small fix in serialization warmup code