diff --git a/lib/clore_partner.py b/lib/clore_partner.py
index 5f95c6e..20d397a 100644
--- a/lib/clore_partner.py
+++ b/lib/clore_partner.py
@@ -257,7 +257,7 @@ async def check_to_pull_selftest(current_specs):
             if nvidia_gpu["pcie_width"] < min_width:
                 min_width =  nvidia_gpu["pcie_width"]
                 if " MiB" in nvidia_gpu["mem_total"]:
-                    gpu_total_vram += int(nvidia_gpu["mem_total"])
+                    gpu_total_vram += int(nvidia_gpu["mem_total"].replace(" MiB", ''))
         if gpu_name in auto_pull_selftest_gpus and current_specs["ram"] > 7 and int(current_specs["cpus"].split('/')[0]) >= 4 and not mixed_cards and min_width > 1 and gpu_total_vram < current_specs["ram"] * 1024 and float(current_specs["disk"].split(' ')[-1].replace("GB", '')) > 25:
             await utils.async_run_command("docker pull vastai/test:selftest", 14400, non_interactive_env)
     except Exception as e: