diff --git a/package-lock.json b/package-lock.json
index 31d1112..f396429 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -9,6 +9,7 @@
       "version": "2.4.0",
       "license": "Apache-2.0",
       "dependencies": {
+        "argparse": "^2.0.1",
         "gl": "^6.0.2",
         "log4js": "^6.9.1",
         "molstar": "~4.5.0",
@@ -20,6 +21,7 @@
         "pdb2cif": "lib/cli/pdb2cif.js"
       },
       "devDependencies": {
+        "@types/argparse": "^2.0.16",
         "@types/gl": "^6.0.2",
         "@types/jest": "^29.5.0",
         "@types/jpeg-js": "^0.3.7",
diff --git a/package.json b/package.json
index de026eb..a9b7a65 100644
--- a/package.json
+++ b/package.json
@@ -33,6 +33,7 @@
   },
   "homepage": "https://github.com/PDBeurope/pdb-images#readme",
   "devDependencies": {
+    "@types/argparse": "^2.0.16",
     "@types/gl": "^6.0.2",
     "@types/jest": "^29.5.0",
     "@types/jpeg-js": "^0.3.7",
@@ -50,6 +51,7 @@
     "typescript": "^5.0.3"
   },
   "dependencies": {
+    "argparse": "^2.0.1",
     "gl": "^6.0.2",
     "log4js": "^6.9.1",
     "molstar": "~4.5.0",
diff --git a/src/save.ts b/src/save.ts
index bd10645..b3ac7ce 100644
--- a/src/save.ts
+++ b/src/save.ts
@@ -42,7 +42,7 @@ export function makeSaveFunction(plugin: HeadlessPluginContext, outDir: string,
             if (args.render_each_size || !fullsizeImage) {
                 // Render new image
                 plugin.canvas3d!.commit(true);
-                image = await plugin.renderer.getImageRaw(size, postprocessing);
+                image = await plugin.getImageRaw(size, postprocessing);
                 if (!args.no_axes) {
                     addAxisIndicators(image, spec._view);
                 }