Fix "Undefined variable" notice. Thanks wet.