Fix transaction stages feature