diff --git a/.travis.yml b/.travis.yml
index cdbf30ec..6d788546 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,3 +1,5 @@
+language: python
+python: 3.6
 sudo: required
 
 branches:
@@ -11,6 +13,9 @@ env:
   - CHANGE_MINIKUBE_NONE_USER=true
 
 before_script:
+  # lint Python for syntax errors and undefined names
+  - pip install flake8
+  - flake8 . --count --select=E901,E999,F821,F822,F823 --show-source --statistics
   # install Go
   - (sudo add-apt-repository -y ppa:gophers/archive; sudo apt-get -y update; sudo apt-get -y install golang-1.9-go) > /dev/null 2>&1
   - export PATH=/usr/lib/go-1.9/bin:$PATH; export GOROOT=/usr/lib/go-1.9/
diff --git a/etc/examples/c10d-dist-onnx/model-files/train_dist_onnx.py b/etc/examples/c10d-dist-onnx/model-files/train_dist_onnx.py
index cc0859a5..b75a1645 100644
--- a/etc/examples/c10d-dist-onnx/model-files/train_dist_onnx.py
+++ b/etc/examples/c10d-dist-onnx/model-files/train_dist_onnx.py
@@ -117,7 +117,7 @@ def run(rank, size, batch_size, is_gpu):
             if not (size == 1):
                 average_gradients(model)
             optimizer.step()
-        print('Process ', rank=dist.get_rank(),
+        print('Process ', dist.get_rank(),
               ', epoch ', epoch, '. avg_loss: ',
               epoch_loss / len(train_set))
 
diff --git a/etc/examples/c10d-onnx-mpi/model-files/train_dist_onnx_mpi.py b/etc/examples/c10d-onnx-mpi/model-files/train_dist_onnx_mpi.py
index 5a170dd3..830b8180 100644
--- a/etc/examples/c10d-onnx-mpi/model-files/train_dist_onnx_mpi.py
+++ b/etc/examples/c10d-onnx-mpi/model-files/train_dist_onnx_mpi.py
@@ -90,7 +90,7 @@ def run(rank, size, batch_size, is_gpu):
     test_set = torch.utils.data.DataLoader(
         test_set, batch_size=batch_size, shuffle=True, pin_memory=True)
 
-    num_batches = ceil(len(train_set.dataset) / float(bsz))
+    num_batches = ceil(len(train_set.dataset) / float(batch_size))
     # To train model
     model.train()
     for epoch in range(100):