update tests for latest code