small style improvment