Entity bug fixes