diff --git a/examples/pinn_forward/Burgers.py b/examples/pinn_forward/Burgers.py
index 893e70789..7b6883ed1 100644
--- a/examples/pinn_forward/Burgers.py
+++ b/examples/pinn_forward/Burgers.py
@@ -1,4 +1,5 @@
 """Backend supported: tensorflow.compat.v1, tensorflow, pytorch, paddle"""
+
 import deepxde as dde
 import numpy as np
 
@@ -38,13 +39,12 @@ def pde(x, y):
 model.train(iterations=15000)
 model.compile("L-BFGS")
 losshistory, train_state = model.train()
-dde.saveplot(losshistory, train_state, issave=True, isplot=True)
 # """Backend supported: pytorch"""
 # # Run NNCG after Adam and L-BFGS
 # dde.optimizers.set_NNCG_options(rank=50, mu=1e-1)
 # model.compile("NNCG")
 # losshistory, train_state = model.train(iterations=1000, display_every=100)
-# dde.saveplot(losshistory, train_state, issave=True, isplot=True)
+dde.saveplot(losshistory, train_state, issave=True, isplot=True)
 
 X, y_true = gen_testdata()
 y_pred = model.predict(X)