Fixed compile errors from the last commit.