Changing some tests to adapt to backend prediction changes