diff --git a/juju-charms/layers/simple/actions/touch b/juju-charms/layers/simple/actions/touch
index d85d3fa..7e30af4 100755
--- a/juju-charms/layers/simple/actions/touch
+++ b/juju-charms/layers/simple/actions/touch
@@ -18,15 +18,14 @@
 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()