Fix NameError.
Improve readability.