Remvoing Stateful from Workflow (with modified migration)