1 files changed, 5 insertions, 4 deletions
diff --git a/src/expr/expr.h b/src/expr/expr.h
index 27b7846db..6c615d391 100644
--- a/src/expr/expr.h
+++ b/src/expr/expr.h
@@ -28,7 +28,8 @@
 namespace CVC4 {
 
 // The internal expression representation
-class Node;
+template <bool count_ref>
+class NodeTemplate;
 
 /**
  * Class encapsulating CVC4 expressions and methods for constructing new
@@ -161,10 +162,10 @@ protected:
    * @param em the expression manager that handles this expression
    * @param node the actual expression node pointer
    */
-  Expr(ExprManager* em, Node* node);
+  Expr(ExprManager* em, NodeTemplate<true>* node);
 
   /** The internal expression representation */
-  Node* d_node;
+  NodeTemplate<true>* d_node;
 
   /** The responsible expression manager */
   ExprManager* d_exprManager;
@@ -173,7 +174,7 @@ protected:
    * Returns the actual internal node.
    * @return the internal node
    */
-  Node getNode() const;
+  NodeTemplate<true> getNode() const;
 
   // Friend to access the actual internal node information and private methods
   friend class SmtEngine;