minor bug fixing