1 files changed, 12 insertions, 10 deletions
diff --git a/include/gras/element.i b/include/gras/element.i
index 1408eac..be3814e 100644
--- a/include/gras/element.i
+++ b/include/gras/element.i
@@ -34,23 +34,25 @@
 ////////////////////////////////////////////////////////////////////////
 // Operator overloads for Element
 ////////////////////////////////////////////////////////////////////////
-%extend gras::Element
+%{
+
+inline bool gras_element_equal(const gras::Element &lhs, const gras::Element &rhs)
 {
-    std::string __str__(void) const
-    {
-       return ($self)->to_string();
-    }
+    return lhs.get() == rhs.get();
+}
 
-    bool _equal(const Element &rhs) const
-    {
-        return ($self)->get() == rhs.get();
-    }
+%}
 
+%extend gras::Element
+{
     %insert("python")
     %{
         def __eq__(self, rhs):
             if not isinstance(rhs, Element): return False
-            return self._equal(rhs)
+            return gras_element_equal(self, rhs)
+
+        def __str__(self):
+            return self.to_string()
     %}
 }