Fix issue with variable