Fixed a bug :)