diff --git a/package-lock.json b/package-lock.json
index fbbda07..0974957 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "tree-sitter-metamodelica",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "tree-sitter-metamodelica",
-      "version": "0.2.0",
+      "version": "0.2.1",
       "hasInstallScript": true,
       "license": "SEE LICENSE IN OSMC-License.txt",
       "dependencies": {
diff --git a/package.json b/package.json
index f365f8e..d52c0e5 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "tree-sitter-metamodelica",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "MetaModelica grammar for tree-sitter",
   "main": "index.js",
   "types": "bindings/node",
diff --git a/rules/a4-equations.js b/rules/a4-equations.js
index 558a794..eabc1af 100644
--- a/rules/a4-equations.js
+++ b/rules/a4-equations.js
@@ -61,7 +61,6 @@ module.exports = {
     repeat($._constraint_annotation_list)
   )),
 
-  // TODO: Do look ahead
   _equation_annotation_list: $ => choice(
     seq(
       $.equation,
@@ -73,11 +72,9 @@ module.exports = {
     )
   ),
 
-  // TODO: What the heck is `{ ... }?` ?
-  // TODO: Do look ahead
   _constraint_annotation_list: $ => choice(
     seq(
-      $.equation,
+      $.constraint,
       $._SEMICOLON
     ),
     seq(