Another fix for the size check

Another fix for the size check