Fix test build failure