updated test to reflect recent changes