diff --git a/python/paddle/v2/framework/tests/test_nce.py b/python/paddle/v2/fluid/tests/test_nce.py
similarity index 86%
rename from python/paddle/v2/framework/tests/test_nce.py
rename to python/paddle/v2/fluid/tests/test_nce.py
index 8b1e7a6bb535fe00e426feaf3d8a22840660f03c..82978f2d230a3fa44faeb07c3aade75d115e5a26 100644
--- a/python/paddle/v2/framework/tests/test_nce.py
+++ b/python/paddle/v2/fluid/tests/test_nce.py
@@ -55,10 +55,10 @@ class TestNCE(OpTest):
             'sampled_labels': range(num_sampled_classes)
         }
         self.inputs = {
-            'X': input,
+            'Input': input,
             'Label': labels,
-            'W': weight,
-            'B': bias,
+            'Weight': weight,
+            'Bias': bias,
             'SampleWeight': sample_weight
         }
 
@@ -66,11 +66,12 @@ class TestNCE(OpTest):
         self.generate_data(5, 5, 4, 1, 2)
 
     def compute(self):
-        out = nce(self.inputs['X'], self.inputs['W'], self.inputs['B'],
-                  self.inputs['SampleWeight'], self.inputs['Label'],
-                  self.attrs['num_classes'], self.attrs['num_sampled_classes'])
+        out = nce(self.inputs['Input'], self.inputs['Weight'],
+                  self.inputs['Bias'], self.inputs['SampleWeight'],
+                  self.inputs['Label'], self.attrs['num_classes'],
+                  self.attrs['num_sampled_classes'])
         self.outputs = {
-            'Out': out[0],
+            'Cost': out[0],
             'SampleLogits': out[1],
             'SampleLabels': out[2]
         }
@@ -84,7 +85,8 @@ class TestNCE(OpTest):
         self.check_output()
 
     def test_check_grad(self):
-        self.check_grad(["X", "W", "B"], "Out", max_relative_error=0.02)
+        self.check_grad(
+            ["Input", "Weight", "Bias"], "Cost", max_relative_error=0.02)
 
 
 class TestNCECase1(TestNCE):