diff --git a/openssh/known_hosts.sls b/openssh/known_hosts.sls
index af22f6b..9541e55 100644
--- a/openssh/known_hosts.sls
+++ b/openssh/known_hosts.sls
@@ -1,14 +1,9 @@
 {% from "openssh/map.jinja" import openssh with context %}
 
-check for existing dig:
-  cmd.run:
-    - name: which dig
-
 ensure dig is available:
   pkg.installed:
     - name: {{ openssh.dig_pkg }}
-    - onfail:
-      - cmd: check for existing dig
+    - unless: which dig
 
 manage ssh_known_hosts file:
   file.managed: