Fix distribution generation