import sys
sys.path.append('lib')
-from charms.reactive import main
-from charms.reactive import set_state
+from charms.reactive import main, set_flag
from charmhelpers.core.hookenv import action_fail, action_name
"""
`set_state` only works here because it's flushed to disk inside the `main()`
loop. remove_state will need to be called inside the action method.
"""
-set_state('actions.{}'.format(action_name()))
+set_flag('actions.{}'.format(action_name()))
try:
main()