diff --git a/src/swagger/swagger_generator.py b/src/swagger/swagger_generator.py
index b3734264..e3204dfe 100644
--- a/src/swagger/swagger_generator.py
+++ b/src/swagger/swagger_generator.py
@@ -4,6 +4,7 @@
 import yaml
 
 from common.config import config
+from src.resources.entities.entity_map import endpoints
 
 
 class Parameter(object):
@@ -224,9 +225,6 @@ def get_spec_as_dict(self):
 class SwaggerGenerator(object):
     FILE_PATH = Path.cwd() / "src" / "swagger" / "openapi.yaml"
 
-    def __init__(self):
-        self.endpoints = []
-
     @staticmethod
     def pascal_to_normal(input):
         """
@@ -238,18 +236,6 @@ def pascal_to_normal(input):
         words = re.findall(r"[A-Z]?[a-z]+|[A-Z]{2,}(?=[A-Z][a-z]|\d|\W|$)|\d+", input)
         return " ".join(map(str.lower, words))
 
-    def resource_wrapper(self):
-        """
-        Wrapper for Resource classes that appends the class name to the endpoints list
-        """
-
-        def decorate(cls):
-            if config.is_generate_swagger():
-                self.endpoints.append(cls.__name__)
-            return cls
-
-        return decorate
-
     def write_swagger_spec(self):
         """
         Writes the openapi.yaml file
@@ -257,7 +243,7 @@ def write_swagger_spec(self):
         """
         if config.is_generate_swagger():
             swagger_spec = SwaggerSpecification()
-            for endpoint in self.endpoints:
+            for endpoint in endpoints:
                 entity = Entity(endpoint)
                 swagger_spec.add_path(entity.entity_count_endpoint)
                 swagger_spec.add_path(entity.entity_id_endpoint)