fix a bug in support set learning