diff --git a/paddle/fluid/imperative/layer.h b/paddle/fluid/imperative/layer.h
index 8d4fac6bcbc280062a8b8296b19b1e790b8f6385..84f100fd6031768fbfce12e3fdc639655375493d 100644
--- a/paddle/fluid/imperative/layer.h
+++ b/paddle/fluid/imperative/layer.h
@@ -136,8 +136,6 @@ class VarBase {
 
  public:
   virtual ~VarBase() {
-    // LOG(ERROR) << "remove var " << name_;
-
     if (block_ && !persistable_) {
       block_->RemoveVar(name_);
     }
diff --git a/python/paddle/fluid/imperative/tracer.py b/python/paddle/fluid/imperative/tracer.py
index 8b53d6c2822d3c9a58529e0efed549a160375b7d..1064ad63e7103acde9bb8106b7791441ce68849b 100644
--- a/python/paddle/fluid/imperative/tracer.py
+++ b/python/paddle/fluid/imperative/tracer.py
@@ -24,10 +24,6 @@ __all__ = ['Tracer']
 
 
 def release_op(op):
-    import gc
-    assert len(
-        gc.get_referrers(framework._imperative_tracer()._ops[
-            op._trace_id])) == 1
     del framework._imperative_tracer()._ops[op._trace_id]
 
 
@@ -59,6 +55,8 @@ class Tracer(core.Tracer):
             if len(backward_refs) > 0:
                 op.iop.register_backward_hooks(release_op)
 
+                # TODO(minqiyang): remove all inputs and outputs after seperate
+                # var and grad
                 op.backward_refs = defaultdict(list)
                 for k, v in six.iteritems(op.inputs):
                     if k in backward_refs:
diff --git a/python/paddle/fluid/tests/unittests/test_imperative.py b/python/paddle/fluid/tests/unittests/test_imperative_basic.py
similarity index 100%
rename from python/paddle/fluid/tests/unittests/test_imperative.py
rename to python/paddle/fluid/tests/unittests/test_imperative_basic.py