Fixed issue where entity wasn't updating properly