Small but important fix