diff --git a/pkgs/build-support/writers/scripts.nix b/pkgs/build-support/writers/scripts.nix
index bceac1b0c959..edc61403dfa7 100644
--- a/pkgs/build-support/writers/scripts.nix
+++ b/pkgs/build-support/writers/scripts.nix
@@ -597,7 +597,7 @@ rec {
   # writePython3 "test_python3" { libraries = [ pkgs.python3Packages.pyyaml ]; } ''
   #   import yaml
   #
-  #   y = yaml.load("""
+  #   y = yaml.safe_load("""
   #     - test: success
   #   """)
   #   print(y[0]['test'])
@@ -615,7 +615,7 @@ rec {
   # writePyPy3 "test_pypy3" { libraries = [ pkgs.pypy3Packages.pyyaml ]; } ''
   #   import yaml
   #
-  #   y = yaml.load("""
+  #   y = yaml.safe_load("""
   #     - test: success
   #   """)
   #   print(y[0]['test'])