fixed bug....