Find assembly correctly in pyspark