fixed a bug..