Fix for missing import in block.py.