fixed bug..