diff --git a/hapi/test/dist_mnist.py b/hapi/test/dist_mnist.py
index 13e39994cfe4b68cd5e6a80056c72961f5bc85b9..ea407a8b3d1445b5a7b8ee9aaa634f2a5fb0ad12 100644
--- a/hapi/test/dist_mnist.py
+++ b/hapi/test/dist_mnist.py
@@ -24,11 +24,9 @@ import contextlib
 
 import paddle
 from paddle import fluid
-from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear
-from paddle.io import BatchSampler, DataLoader
 
 from hapi.model import Model, Input, set_device
-from hapi.loss import Loss, CrossEntropy
+from hapi.loss import CrossEntropy
 from hapi.vision.models import LeNet
 from hapi.metrics import Accuracy
 from hapi.callbacks import ProgBarLogger