Fixing issue with code not showing in doc