Fix  bugs