diff --git a/src/Orbit-wise.jl b/src/Orbit-wise.jl
index afeef29..4afa743 100644
--- a/src/Orbit-wise.jl
+++ b/src/Orbit-wise.jl
@@ -37,11 +37,13 @@ function OrbitData(sett::Settings)
    splap² = similar(splap)
    splap² = GroupRings.mul!(splap², splap, splap, pm);
 
-   # Uπs = load(joinpath(name, "U_pis.jld"), "Uπs");
    Uπs = load(joinpath(prepath(sett), "U_pis.jld"), "Uπs")
+   nzros = [i for i in 1:length(Uπs) if size(Uπs[i],2) !=0]
+   Uπs = Uπs[nzros]
    Uπs = sparsify!.(Uπs, sett.tol, check=true, verbose=true)
+
    #dimensions of the corresponding πs:
-   dims = load(joinpath(prepath(sett), "U_pis.jld"), "dims")
+   dims = load(joinpath(prepath(sett), "U_pis.jld"), "dims")[nzros]
 
    m, P = init_model(size(Uπs,1), [size(U,2) for U in Uπs]);
 
diff --git a/src/OrbitDecomposition.jl b/src/OrbitDecomposition.jl
index ce582cf..07933a8 100644
--- a/src/OrbitDecomposition.jl
+++ b/src/OrbitDecomposition.jl
@@ -137,12 +137,7 @@ function perm_reps(S::Vector, autS::Group, radius::Int)
 end
 
 function reconstruct_sol{T<:GroupElem, S<:Nemo.perm}(preps::Dict{T, S},
-   aUs::Vector, aPs::Vector, adims::Vector)
-
-   idx = [π for π in 1:length(aUs) if size(aUs[π], 2) != 0]
-   Us = aUs[idx]
-   Ps = aPs[idx]
-   dims = adims[idx];
+   Us::Vector, Ps::Vector, dims::Vector)
 
    l = length(Us)
    transfP = [dims[π].*Us[π]*Ps[π]*Us[π]' for π in 1:l]