fix bugs =)