fixed bug!!! :)