update fix to tests