Minor edit to bug fix