Fix minor minor bug