Fix missing array issue