Fix for last commit

sz was not defined in the previous commit, whoops!