fixed statefulHoist to return StatefulMorph instead of a function