Fix test compilation.

Mathias, can you have a look and confirm that this is correct?