Fix mktheoryrewriter and mktheorytraits for nullaryoperator.

2 files changed, 13 insertions, 0 deletions

diff --git a/src/theory/mkrewriter b/src/theory/mkrewriter
index ba623956f..ea7deb270 100755
--- a/src/theory/mkrewriter
+++ b/src/theory/mkrewriter
@@ -212,6 +212,12 @@ function constant {
   check_theory_seen
 }
 
+function nullaryoperator {
+  # nullaryoperator K ["comment"]
+  lineno=${BASH_LINENO[0]}
+  check_theory_seen
+}
+
 function check_theory_seen {
   if $seen_endtheory; then
     echo "$kf:$lineno: error: command after \"endtheory\" declaration (endtheory has to be last)" >&2
diff --git a/src/theory/mktheorytraits b/src/theory/mktheorytraits
index 983cfade4..15166fc1f 100755
--- a/src/theory/mktheorytraits
+++ b/src/theory/mktheorytraits
@@ -345,6 +345,13 @@ function constant {
   register_kind "$1" 0 "$5"
 }
 
+function nullaryoperator {
+  # nullaryoperator K ["comment"]
+  lineno=${BASH_LINENO[0]}
+  check_theory_seen
+  register_kind "$1" 0 "$2"
+}
+
 function register_sort {
   id=$1
   cardinality=$2