Fix compile bug (thanks @chrisdone)