Fixed wrong variable name in unittest.py