Modified access of the returned array

Changed the address to the index of the returned array