1 files changed, 9 insertions, 2 deletions
diff --git a/src/theory/theoryof_table_middle.h b/src/theory/theoryof_table_middle.h
index a0586d0ce..54be78b95 100644
--- a/src/theory/theoryof_table_middle.h
+++ b/src/theory/theoryof_table_middle.h
@@ -28,8 +28,9 @@ public:
   TheoryOfTable() :
     d_table(new Theory*[::CVC4::kind::LAST_KIND]) {
   }
-  ~TheoryOfTable(){
-    delete[] d_table;
+
+  ~TheoryOfTable() {
+    delete [] d_table;
   }
 
   Theory* operator[](TNode n) {
@@ -37,3 +38,9 @@ public:
            "illegal to inquire theoryOf(UNDEFINED_KIND or out-of-range)");
     return d_table[n.getKind()];
   }
+
+  Theory* operator[](::CVC4::Kind k) {
+    Assert(k >= 0 && k < ::CVC4::kind::LAST_KIND,
+           "illegal to inquire theoryOf(UNDEFINED_KIND or out-of-range)");
+    return d_table[k];
+  }