1 files changed, 7 insertions, 2 deletions
diff --git a/tikzit/src/gui/nodeitem.h b/tikzit/src/gui/nodeitem.h
index 60b2c05..867d8a3 100644
--- a/tikzit/src/gui/nodeitem.h
+++ b/tikzit/src/gui/nodeitem.h
@@ -4,13 +4,18 @@
 #include "node.h"
 
 #include <QObject>
-#include <QGraphicsEllipseItem>
+#include <QGraphicsItem>
+#include <QPainterPath>
+#include <QRectF>
 
-class NodeItem : public QGraphicsEllipseItem
+class NodeItem : public QGraphicsItem
 {
 public:
     NodeItem(Node *node);
     void syncPos();
+    void paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget);
+    QPainterPath shape() const;
+    QRectF boundingRect() const;
 private:
     Node *_node;
 };