fix bugs..