diff --git a/cmake/PostInstall.in b/cmake/PostInstall.in
index c0b657c..be8ce76 100644
--- a/cmake/PostInstall.in
+++ b/cmake/PostInstall.in
@@ -1,3 +1,9 @@
+if [ $1 == 2 ]; then
+    systemctl stop tsg-diagnose
+    docker rmi -f badssl-tsg-diagnose
+    docker rmi -f unittest-tsg-diagnose
+    docker rmi -f wpr-tsg-diagnose
+fi
 docker load < /opt/tsg/tsg-diagnose/images/badssl.tar
 docker load < /opt/tsg/tsg-diagnose/images/unittest.tar
 docker load < /opt/tsg/tsg-diagnose/images/wpr.tar
\ No newline at end of file
diff --git a/cmake/PostunInstall.in b/cmake/PostunInstall.in
index 2132aa5..3fc02fc 100644
--- a/cmake/PostunInstall.in
+++ b/cmake/PostunInstall.in
@@ -1,3 +1,5 @@
-docker rmi -f badssl-tsg-diagnose
-docker rmi -f unittest-tsg-diagnose
-docker rmi -f wpr-tsg-diagnose
\ No newline at end of file
+if [ $1 == 0 ]; then
+    docker rmi -f badssl-tsg-diagnose
+    docker rmi -f unittest-tsg-diagnose
+    docker rmi -f wpr-tsg-diagnose
+fi
\ No newline at end of file
diff --git a/cmake/PreunInstall.in b/cmake/PreunInstall.in
index 861ed3f..18a38a6 100644
--- a/cmake/PreunInstall.in
+++ b/cmake/PreunInstall.in
@@ -1 +1,3 @@
-systemctl stop tsg-diagnose
\ No newline at end of file
+if [ $1 == 0 ]; then
+    systemctl stop tsg-diagnose
+fi
\ No newline at end of file