Fixed a small issued regarding the fix to read the whole schema