diff --git a/paddle/api/GradientMachine.cpp b/paddle/api/GradientMachine.cpp
index 66115f8293b905809639afff779abfdb2bb3a54e..a44763bfa53ed5689d665cbd78f09417ec7de6da 100644
--- a/paddle/api/GradientMachine.cpp
+++ b/paddle/api/GradientMachine.cpp
@@ -146,7 +146,7 @@ void GradientMachine::randParameters() { m->machine->randParameters(); }
 
 Matrix* GradientMachine::getLayerOutput(const std::string& layerName) const
     throw(UnsupportError) {
-  auto nn = std::dynamic_pointer_cast<paddle::NeuralNetwork>(m->machine);
+  auto nn = m->machine;
   if (nn) {
     auto mat = nn->getLayerOutput(layerName);
     return Matrix::createByPaddleMatrixPtr(&mat);