update tests to fix build error