Fix compile issue from last commit. Oops. :-(