Small update fix