diff --git a/pipenv/patched/safety/__main__.py b/pipenv/patched/safety/__main__.py
index 90e702a7..cbd1a075 100644
--- a/pipenv/patched/safety/__main__.py
+++ b/pipenv/patched/safety/__main__.py
@@ -44,8 +44,5 @@ def insert_pipenv_dirs():
 
 if __name__ == "__main__":  # pragma: no cover
     insert_pipenv_dirs()
-    yaml_lib = "pipenv.patched.yaml{0}".format(sys.version_info[0])
-    locals()[yaml_lib] = __import__(yaml_lib)
-    sys.modules["yaml"] = sys.modules[yaml_lib]
     from pipenv.patched.safety.cli import cli
     cli(prog_name="safety")
diff --git a/tasks/vendoring/patches/patched/safety-main.patch b/tasks/vendoring/patches/patched/safety-main.patch
index 3d214863..30073c49 100644
--- a/tasks/vendoring/patches/patched/safety-main.patch
+++ b/tasks/vendoring/patches/patched/safety-main.patch
@@ -2,7 +2,7 @@ diff --git a/pipenv/patched/safety/__main__.py b/pipenv/patched/safety/__main__.
 index d9a0bdab..f905408a 100644
 --- a/pipenv/patched/safety/__main__.py
 +++ b/pipenv/patched/safety/__main__.py
-@@ -1,8 +1,51 @@
+@@ -1,8 +1,48 @@
  """Allow safety to be executable through `python -m safety`."""
  from __future__ import absolute_import
  
@@ -50,8 +50,5 @@ index d9a0bdab..f905408a 100644
  
  if __name__ == "__main__":  # pragma: no cover
 +    insert_pipenv_dirs()
-+    yaml_lib = "pipenv.patched.yaml{0}".format(sys.version_info[0])
-+    locals()[yaml_lib] = __import__(yaml_lib)
-+    sys.modules["yaml"] = sys.modules[yaml_lib]
 +    from safety.cli import cli
      cli(prog_name="safety")