fixed bug and working fine

fixed a stupid bug