fix build failure in test