Fix test so that it maybe compiles