Fixed bug :)