Update test for new code