Fix build error in test