Correct variable issue

Hi Aaron I think it works better this way.