diff --git a/irrep.jl b/irrep.jl
index 36d8d8d..beb00a5 100644
--- a/irrep.jl
+++ b/irrep.jl
@@ -1,4 +1,4 @@
-using DelimitedFiles, LinearAlgebra, StatsBase
+using DelimitedFiles, LinearAlgebra
 
 function calculate_all_data(N::Int)
     ks = -N÷2:N÷2-1
@@ -35,7 +35,7 @@ function calculate_A1_data(N::Int)
     end
 
     unique_i = unique(labels)
-    multiplicity = [countmap(labels)[i] for i in unique_i]
+    multiplicity = [count(labels.==i) for i in unique_i]
     unique_point = transpose(lattice[:, unique_i])
 
     return unique_i, unique_point, multiplicity