Fix UndefVarError

Created by: michel2323

• Fixed UndefVarError introduced in https://github.com/JuliaGPU/KernelAbstractions.jl/pull/236

Moving from v0.6.0 to v0.6.1 triggered a few undefined variable errors. This PR fixes those.