Fix compile error

Oops!