Fix AttributeError introduced in 0.3.4