fix bugs....