Fixed compile errors from the previous commit.