Fixed up a bit of compile error