Fixed build error in tests