Fixed a "simplification" I added without testing. -_-