Index: container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java
===================================================================
--- container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java	(revision 785536)
+++ container/openejb-core/src/main/java/org/apache/openejb/core/stateful/StatefulContainer.java	(working copy)
@@ -846,6 +846,7 @@
 
             // Invoke afterBegin
             ThreadContext callContext = new ThreadContext(instance.deploymentInfo, instance.primaryKey, Operation.AFTER_BEGIN);
+            callContext.setCurrentAllowedStates(StatefulContext.getStates());
             ThreadContext oldCallContext = ThreadContext.enter(callContext);
             try {