Fix minor bug
Fix minor bug #2