diff --git a/llvm/projects/soc_simulator/src/driver.py b/llvm/projects/soc_simulator/src/driver.py
index 4018477ac755ff839ed4509a4a78341ce30251e1..d76dbb5150a0fa1585426e8ab818cdc3d1e00024 100644
--- a/llvm/projects/soc_simulator/src/driver.py
+++ b/llvm/projects/soc_simulator/src/driver.py
@@ -1,5 +1,3 @@
-# Python driver -- ported from Perl driver (driver.pl)
-
 from collections import defaultdict
 import os
 import subprocess
@@ -174,7 +172,7 @@ def run_promise_simulation(swing, layer_data):
     total_time_energy = output.strip().split(',')
 
     assert(len(total_time_energy) == 2)
-    return total_time_energy[0], total_time_energy[1]
+    return float(total_time_energy[0]), float(total_time_energy[1])
 
 
 def run_gpu_simulation(curr_layer, layer_name, tensor_ind):
@@ -255,4 +253,4 @@ if __name__ == "__main__":
     '''
     parse_tensor_layer_file("/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_mobilenet/mobilenet_layers.txt")
     parse_tensor_table("/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_pldi/mobilenet_results/mobilenet_tensors.txt")
-    run_simulations("/home/nvidia/Gitlab/hpvm/llvm/projects/soc_simulator/pipeline_GEMO/pipeline_GEMO_promise_confs1.txt", "blah")
+    run_simulations("/home/nvidia/Gitlab/hpvm/llvm/projects/hpvm-tensor-rt/build_mobilenet/mobilenet_conf2.txt", "blah")