Updades to tests