Fixed a minor bug in this thing.