Missing batch ctor
Fix type hint