Fix compile error

Oops. That's what happens when I don't test properly...