fixed a bug...