Fix PySpark for assembly run and include it in dist