diff --git a/realis_admin.go b/realis_admin.go
index f2e99f2..f2759ea 100644
--- a/realis_admin.go
+++ b/realis_admin.go
@@ -59,10 +59,7 @@ func (c *Client) SLADrainHosts(policy *aurora.SlaPolicy, timeout int64, hosts ..
 		return nil, errors.New("no hosts provided to drain")
 	}
 
-	if policy == nil ||
-		(policy.PercentageSlaPolicy == nil &&
-			policy.CountSlaPolicy == nil &&
-			policy.CoordinatorSlaPolicy == nil) {
+	if policy == nil || policy.CountSetFieldsSlaPolicy() == 0 {
 		policy = &defaultSlaPolicy
 		c.logger.Printf("Warning: start draining with default sla policy %v", policy)
 	}