Fix test so it compiles