minor bug fixing
minor bug fixing