bugs fixed

minor bug