typo in update_layers in model2