Fix updateLayout calling