Fixing the size check