Fixed compile.

Had forgotten to edit two lines.