Changing test to fit backend changes