Change entity_load() to entity_load_single()