diff --git a/api/onnx_web/chain/correct_gfpgan.py b/api/onnx_web/chain/correct_gfpgan.py
index 203c8375..e25ac21e 100644
--- a/api/onnx_web/chain/correct_gfpgan.py
+++ b/api/onnx_web/chain/correct_gfpgan.py
@@ -27,7 +27,7 @@ def load_gfpgan(
 
     if upsampler is None:
         bg_upscale = upscale.rescale(upscale.outscale)
-        upsampler = load_resrgan(ctx, bg_upscale)
+        upsampler = load_resrgan(ctx, bg_upscale, device)
 
     face_path = path.join(ctx.model_path, "%s.pth" % (upscale.correction_model))