Fixed syntax to be python3 compatible