Update to tests