fixed bug ;)