Fixed build error in test