Slight Revision

I fixed a minor bug I found in the last commit.