diff --git a/no_label_milestone.php b/no_label_milestone.php
index 808deb9..3029fd5 100644
--- a/no_label_milestone.php
+++ b/no_label_milestone.php
@@ -39,9 +39,9 @@ $callback=function($issue) use ($nolabelLabel,$nomilestoneLabel){
     $issue->removeLabel( $nolabelLabel );
   }
   //do the no-milestone thing
-  if( count($issue->getData()->milestone)==0 && !$issue->hasLabel($nomilestoneLabel) ){
+  if( !$issue->getData()->milestone && !$issue->hasLabel($nomilestoneLabel) ){
     $issue->addLabel( $nomilestoneLabel );
-  }elseif( count($issue->getData()->milestone)>0 && $issue->hasLabel($nomilestoneLabel) ){
+  }elseif( $issue->getData()->milestone && $issue->hasLabel($nomilestoneLabel) ){
     $issue->removeLabel( $nomilestoneLabel );
   }
 };