Fixed a bug, oops