Fixed a minor bug (probably)