fix minor bug 2