Fix wrong class name in test_engine.py.