1 files changed, 6 insertions, 9 deletions
diff --git a/src/prop/cnf_stream.cpp b/src/prop/cnf_stream.cpp
index 44768009e..611689c2b 100644
--- a/src/prop/cnf_stream.cpp
+++ b/src/prop/cnf_stream.cpp
@@ -113,15 +113,12 @@ SatLiteral TseitinCnfStream::handleAtom(TNode node) {
   bool theoryLiteral = node.getKind() != kind::VARIABLE;
   SatLiteral lit = newLiteral(node, theoryLiteral);
 
-  switch(node.getKind()) {
-  case TRUE:
-    assertClause(lit);
-    break;
-  case FALSE:
-    assertClause(~lit);
-    break;
-  default:
-    break;
+  if(node.getKind() == kind::CONST_BOOLEAN) {
+    if(node.getConst<bool>()) {
+      assertClause(lit);
+    } else {
+      assertClause(~lit);
+    }
   }
 
   return lit;