diff --git a/configs/heterogeneous_c2_simple_noc.yml b/configs/heterogeneous_c2_simple_noc.yml
index 8a3401fe..c9eff44c 100644
--- a/configs/heterogeneous_c2_simple_noc.yml
+++ b/configs/heterogeneous_c2_simple_noc.yml
@@ -8,11 +8,15 @@ core_stats_print_period_cycles: 10000
 num_stonne_per_core: 8
 num_stonne_port: 64
 num_systolic_array_per_core: 2
+core_spad_size_kb: 16384
 
 vpu_num_lanes: 128
 vpu_spad_size_kb_per_lane: 128
 vpu_vector_length_bits: 256
 
+pytorchsim_functional_mode: 1
+pytorchsim_timing_mode: 1
+
 dram_type: ramulator2
 dram_freq_mhz: 940
 dram_channels: 16