fix slight issue