diff --git a/src/pdmgen.py b/src/pdmgen.py
index fe3f826..7487015 100755
--- a/src/pdmgen.py
+++ b/src/pdmgen.py
@@ -645,6 +645,9 @@
         self.countbound = kw.pop('countbound')
         self.op = kw.pop('op')
         self.bound = kw.pop('bound')
+        self.oneshot = TrivialArgument(
+            type='boolean',
+            value=kw.pop('oneshot', False))
         super(CountCondition, self).__init__(**kw)
 
     def setup(self, objs):