update tests for 2.1