Fix for importing Enum in block.py