bug 2 fix