Updated test to reflect changed names