Fix NameError exception caused by recent renaming

Signed-off-by: Niklas Koep <niklas.koep@gmail.com>