Bug fix due to array changes