Update main.py

fixed to work with decisions