diff --git a/softwares/docker-ce.sls b/softwares/docker-ce.sls
index c3634be..b209bba 100644
--- a/softwares/docker-ce.sls
+++ b/softwares/docker-ce.sls
@@ -30,7 +30,7 @@ DockerPrerequisitePackages:
       - ca-certificates
       - gnupg2
       - software-properties-common
-  {% elif grains['os'] == 'CentOS' %}
+  {% elif grains['os_family'] == 'RedHat' %}
       - yum-utils
       - device-mapper-persistent-data
       - lvm2
@@ -50,7 +50,7 @@ DockerCE Repository:
     - file: /etc/apt/sources.list.d/docker-ce.list
     - keyid: 0EBFCD88
     - keyserver: keyserver.ubuntu.com
-  {% elif grains['os'] == 'CentOS'%}
+  {% elif grains['os_family'] == 'RedHat' %}
     - baseurl: {{ LURL }}
     - humanname: docker-ce-stable
     - name: docker-ce
@@ -78,7 +78,7 @@ IwanDocker:
       - docker
   {% if grains['os'] == 'Debian' %}
       - sudo
-  {% elif grains['os'] == 'CentOS' %}
+  {% elif grains['os_family'] == 'RedHat' %}
       - wheel
   {% endif %}