Fix to minor bug