Fixed a bug (been so long that I don't even know what bug it fixes!)