Bug fix (line missed during previous refactoring)