minor bug fixing

minor bug fixing