test updated to new code