fix comparison without pre-train