Fix minimal bugs...