Fixed bug that returned only first item in py4j array loader