diff --git a/qms.py b/qms.py
index 24555998f1c6be0ba047caef2352c9bf70f2522b..c3f430dc7c3b5aefce51cfe5255697ecb3b2479f 100644
--- a/qms.py
+++ b/qms.py
@@ -1,5 +1,10 @@
 import os
 
+def get_category(filename):
+    name_list = filename.split("-")
+
+    return name_list[2]
+
 def prepare_qms(localroot, root, filename, nb):
 
     os.chdir(root)
@@ -15,9 +20,12 @@ def prepare_qms(localroot, root, filename, nb):
     # replace latex command for titles
     filedata = filedata.replace("{-}", "##")
 
+    # get the category
+    category = get_category(filename)
+
     # deal with img location
-    filedata = filedata.replace("policies/"+filename[:-3]+"/", "")
-    filedata = filedata.replace("sops/"+filename[:-3]+"/", "")
+    filedata = filedata.replace("policies/" + category + "/" + filename[:-3] + "/", "")
+    filedata = filedata.replace("sops/" + category + "/" + filename[:-3] + "/", "")
 
     # replace img attributes
     filedata = filedata.replace("{ width=50% }", "")