fix bug 5