diff --git a/sg-backend/db.py b/sg-backend/db.py
index 06dbcd13c3ebcfa5fba7acb3c6533a020b3beee7..fde0325f260fd054b87d86e2b6fd22b63e4ff14b 100644
--- a/sg-backend/db.py
+++ b/sg-backend/db.py
@@ -54,12 +54,13 @@ class Neo4j:
 
     def get_relations(self):
         with self.driver.session() as session:
-            results = session.run("MATCH p = (a)-[r]->(b) RETURN a.name as source,b.name as target, r as relation_type")
+            results = session.run(
+                "MATCH p = (a:Software)-[r]->(b:Software) RETURN a.name as source,b.name as target, r as relation_type")
             return results.data()
 
     def get_nodes(self):
         with self.driver.session() as session:
-            results = session.run("MATCH (a) return Id(a) as key, a.name as name")
+            results = session.run("MATCH (a:Software) return Id(a) as key, a.name as name")
             return results.data()
 
     def insert_software(self, insert_data):
@@ -84,4 +85,4 @@ class Neo4j:
             results = session.run(
                 "MATCH (a:Software), (b:Software) WHERE a.name = '%s' AND b.name = '%s' CREATE (a)-[r:%s]->(b) "
                 "RETURN type(r)" % (insert_data.software1, insert_data.software2, insert_data.relationType))
-            return results.data()
+            return results.data()
\ No newline at end of file