Fix NameError due to module split.