modify class loader impl for three_stage_sim