diff --git a/libgrabsite/main.py b/libgrabsite/main.py
index cfc62a5..fd2a788 100644
--- a/libgrabsite/main.py
+++ b/libgrabsite/main.py
@@ -353,7 +353,7 @@ def main(concurrency, concurrent, delay, recursive, offsite_links, igsets,
 	# only reason to use this is if you're starting wpull manually with modified
 	# arguments, and wpull_hooks.py requires the control files.
 	if which_wpull_command:
-		bin = sys.argv[0].replace("/grab-site", "/wpull") # TODO
+		bin = re.sub("/grab-site$", "/wpull", sys.argv[0])
 		print("GRAB_SITE_WORKING_DIR={} DUPESPOTTER_ENABLED={} {} {}".format(
 			working_dir, int(dupespotter), bin, " ".join(shlex.quote(a) for a in args)))
 		return