Fix code example in README

Without this NameError is thrown