fixed minor robust problem