diff --git a/cuburn/device_code.py b/cuburn/device_code.py
index ba0e23f..441dd24 100644
--- a/cuburn/device_code.py
+++ b/cuburn/device_code.py
@@ -688,7 +688,7 @@ class MWCRNGTest(PTXTest):
     def call_setup(self, ctx):
         # Get current multipliers and seeds from the device
         self.mults = ctx.get_per_thread('mwc_rng_mults', np.uint32)
-        self.fullstates = ctx.get_per_thread('mwc_rng_states', np.uint64)
+        self.fullstates = ctx.get_per_thread('mwc_rng_state', np.uint64)
         self.sums = np.zeros(ctx.nthreads, np.uint64)
 
         print "Running %d states forward %d rounds" % \
@@ -703,7 +703,7 @@ class MWCRNGTest(PTXTest):
         print "Done on host, took %g seconds" % ctime
 
     def call_teardown(self, ctx):
-        dfullstates = ctx.get_per_thread('mwc_rng_states', np.uint64)
+        dfullstates = ctx.get_per_thread('mwc_rng_state', np.uint64)
         if not (dfullstates == self.fullstates).all():
             print "State discrepancy"
             print dfullstates