Fixed a minor bug, but important nonetheless