diff --git a/OrbitDecomposition.jl b/OrbitDecomposition.jl
index fbeb6f2..dca2375 100644
--- a/OrbitDecomposition.jl
+++ b/OrbitDecomposition.jl
@@ -108,7 +108,7 @@ end
 #
 ###############################################################################
 
-function matrix_repr(g::WreathProductElem, E, E_dict)
+function matrix_repr(g::GroupElem, E, E_dict)
    rep_matrix = spzeros(Int, length(E), length(E))
    for (i,e) in enumerate(E)
       j = E_dict[g(e)]