Fix harmless bug
Fix bug in mock generation (class module wasn't included)