Changed import for entity