Fixing Assemble upgrade