Slight change to tests