Updating models to fix elements not being visible