bug fixing
fixed bug