Fix free variable warning