diff --git a/src/finn/transformation/fpgadataflow/make_pynq_driver.py b/src/finn/transformation/fpgadataflow/make_pynq_driver.py
index bfa2fdbf9594c52d9a3a2376312a929b9008c9ea..2c3bd7ee59e23566bbd0acf2241ca67ed2beb3ea 100644
--- a/src/finn/transformation/fpgadataflow/make_pynq_driver.py
+++ b/src/finn/transformation/fpgadataflow/make_pynq_driver.py
@@ -143,7 +143,7 @@ class MakePYNQDriver(Transformation):
             )
             i_tensor_shape_packed = i_tensor_dummy_packed.shape
             # append all input tensor info to relevant lists
-            idt.append(str(i_tensor_dt))
+            idt.append("DataType['%s']" % i_tensor_dt.name)
             ishape_normal.append(i_tensor_shape_normal)
             ishape_folded.append(i_tensor_shape_folded)
             ishape_packed.append(i_tensor_shape_packed)
@@ -190,7 +190,7 @@ class MakePYNQDriver(Transformation):
             )
             o_tensor_shape_packed = o_tensor_dummy_packed.shape
             # append all output tensor info to relevant lists
-            odt.append(str(o_tensor_dt))
+            odt.append("DataType['%s']" % o_tensor_dt.name)
             oshape_normal.append(o_tensor_shape_normal)
             oshape_folded.append(o_tensor_shape_folded)
             oshape_packed.append(o_tensor_shape_packed)
@@ -240,11 +240,11 @@ class MakePYNQDriver(Transformation):
         driver = template_driver.pynq_driver_template
 
         driver = driver.replace("$PLATFORM$", self.platform)
-        driver = driver.replace("$INPUT_FINN_DATATYPE$", str(idt).replace("'", ""))
+        driver = driver.replace("$INPUT_FINN_DATATYPE$", str(idt).replace('"', ""))
         driver = driver.replace("$INPUT_SHAPE_NORMAL$", str(ishape_normal))
         driver = driver.replace("$INPUT_SHAPE_FOLDED$", str(ishape_folded))
         driver = driver.replace("$INPUT_SHAPE_PACKED$", str(ishape_packed))
-        driver = driver.replace("$OUTPUT_FINN_DATATYPE$", str(odt).replace("'", ""))
+        driver = driver.replace("$OUTPUT_FINN_DATATYPE$", str(odt).replace('"', ""))
         driver = driver.replace("$OUTPUT_SHAPE_NORMAL$", str(oshape_normal))
         driver = driver.replace("$OUTPUT_SHAPE_FOLDED$", str(oshape_folded))
         driver = driver.replace("$OUTPUT_SHAPE_PACKED$", str(oshape_packed))