Update align.py

if with_statement is not supported out of the box, other features used will also break it.