diff --git a/plugins/arguments/entity_id.inc b/plugins/arguments/entity_id.inc
index 451d444..5279c15 100644
--- a/plugins/arguments/entity_id.inc
+++ b/plugins/arguments/entity_id.inc
@@ -56,15 +56,11 @@ function ctools_argument_entity_id_context($arg = NULL, $conf = NULL, $empty = F
     return ctools_context_create('entity:' . $entity_type, $arg);
   }
 
-  if (!is_numeric($arg)) {
-    return FALSE;
-  }
-
   $entity = entity_load($entity_type, array($arg));
   if (!$entity) {
     return FALSE;
   }
 
-  return ctools_context_create('entity:' . $entity_type, $entity[$arg]);
+  return ctools_context_create('entity:' . $entity_type, reset($entity));
 }