Fix the umbrella when rebuilding