Little fix to correct variable