diff --git a/bin/gofmtimport b/bin/gofmtimport
index 9c92f60..7fe6636 100755
--- a/bin/gofmtimport
+++ b/bin/gofmtimport
@@ -3,8 +3,16 @@ set -euxo pipefail
 
 [ ${DEBUG:-0} -eq 1 ] && exec 2>/tmp/gofmtimport.dbg
 
-goimports \
+cmdimports=$(command -v gofumports || command -v goimports || true)
+cmdfmt=$(command -v gofumpt || command -v gofmt || true)
+
+[[ -n $cmdimports ]] && [[ -n $cmdfmt ]] || {
+	echo "No imports command or no fmt command found" >&2
+	exit 1
+}
+
+${cmdimports} \
 	-srcdir "$2" \
 	-local "${GOIMPORT_LOCAL:-}" \
 	-w "$2"
-gofmt -s -w "$2"
+${cmdfmt} -s -w "$2"