Fix NameError introduced in 535b041