diff --git a/plugins/arguments/entity_id.inc b/plugins/arguments/entity_id.inc
index 3063fef..476d281 100644
--- a/plugins/arguments/entity_id.inc
+++ b/plugins/arguments/entity_id.inc
@@ -71,13 +71,14 @@ function ctools_argument_entity_id_context($arg = NULL, $conf = NULL, $empty = F
       $match = preg_match('/^id: (\d+)/', $arg, $preg_matches);
     }
 
+    // If we have id which is set by user we create context from it.
     if ($match) {
       $id = $preg_matches[1];
+      if (isset($id) && is_numeric($id)) {
+        return ctools_context_create('entity:' . $entity_type, $id);
+      }
+      return FALSE;
     }
-    if (isset($id) && is_numeric($id)) {
-      return ctools_context_create('entity:' . $entity_type, $id);
-    }
-    return FALSE;
   }
 
   $entities = entity_load($entity_type, array($arg));