updated test to reflect changes