Fix Autoclass error
Fix Autoclass error