diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
index a5411ce4d0b6..361b4f86c038 100644
--- a/python/tvm/relay/frontend/pytorch.py
+++ b/python/tvm/relay/frontend/pytorch.py
@@ -2044,7 +2044,7 @@ def sub(self, inputs, input_types):
             data0, data1, alpha = self.pytorch_promote_types(inputs, input_types)
             return get_relay_op("subtract")(data0, alpha * data1)
         else:
-            data0, data1= self.pytorch_promote_types(inputs, input_types)
+            data0, data1 = self.pytorch_promote_types(inputs, input_types)
             return get_relay_op("subtract")(data0, data1)
 
     def rsub(self, inputs, input_types):