Fix mistake setting variable