Fixed a small bug. Thanks to jkishner for finding it!