Fix minor bug