Add support for ACTION in correction.py

Before and after: http://with-you.pw/6I4Z