Small but important fix

Small but important fix