Fixed the broken entity import