Fixes bug with old code remaining in update script