fixed bug :D