Adding changes to tests