correct bugs ..