Correct bug with variables