diff --git a/ui/src/app/entity/entity-select.directive.js b/ui/src/app/entity/entity-select.directive.js
index 2ac5228..e7ac5fd 100644
--- a/ui/src/app/entity/entity-select.directive.js
+++ b/ui/src/app/entity/entity-select.directive.js
@@ -32,9 +32,11 @@ export default function EntitySelect($compile, $templateCache, entityService) {
var entityTypes = entityService.prepareAllowedEntityTypesList(scope.allowedEntityTypes, scope.useAliasEntityTypes);
- if (entityTypes.length === 1) {
+ var entityTypeKeys = Object.keys(entityTypes);
+
+ if (entityTypeKeys.length === 1) {
scope.displayEntityTypeSelect = false;
- scope.defaultEntityType = entityTypes[0];
+ scope.defaultEntityType = entityTypes[entityTypeKeys[0]];
} else {
scope.displayEntityTypeSelect = true;
}