Fix line mistakenly removed in previous commit