fix not serializable issue