Fixes little variable bug