Fix bugs..