fixed a bug...

I hope