Minor improvment