Fix test build issue