Fixing minor bug, didn't seem to cause any problems