update tests for recent changes