UpdateEntity working again