Fix test build