Fixing build error in tests