Fixed a bug, nothing too serious though