1 files changed, 2 insertions, 2 deletions
diff --git a/src/diagramdocument.h b/src/diagramdocument.h
index 47d1294..c0a63c0 100644
--- a/src/diagramdocument.h
+++ b/src/diagramdocument.h
@@ -27,7 +27,7 @@
 
 class DatabaseTable;
 class DatabaseRelationship;
-class DiagramConnection;
+class Line;
 class DiagramItem;
 class DiagramObject;
 
@@ -81,7 +81,7 @@ public:
 	void itemMoved(DiagramItem *table);
 	void itemHasMoved(DiagramItem *table);
 
-	QList<DiagramConnection *> findConnections(DiagramObject *object);
+	QList<Line *> findConnections(DiagramObject *object);
 	template <class T> QList<T *> itemsByType();
 
 	int gridSize() const;