changed to use class attributes

diff --git a/test/orm/unitofwork.py b/test/orm/unitofwork.py
index 839afe77c11086d9cf67fdbfa0e29e37a68ab860..f28e428ed981c3df5081673b2fad7f6b3290fc75 100644 (file)
--- a/test/orm/unitofwork.py
+++ b/test/orm/unitofwork.py
@@ -482,7 +482,7 @@ class ClauseAttributesTest(UnitOfWorkTest):
         sess.save(u)
         sess.flush()
         assert u.counter == 1
-        u.counter = users_table.c.counter + 1
+        u.counter = User.counter + 1
         sess.flush()
         def go():
             assert u.counter == 2
@@ -498,7 +498,7 @@ class ClauseAttributesTest(UnitOfWorkTest):
         sess.flush()
         assert u.counter == 1
         u.name = 'test2'
-        u.counter = users_table.c.counter + 1
+        u.counter = User.counter + 1
         sess.flush()
         def go():
             assert u.name == 'test2'