diff --git a/Project.toml b/Project.toml
index 4aeaeda..bc560d5 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "FixedEffects"
 uuid = "c8885935-8500-56a7-9867-7708b20db0eb"
-version = "2.0.0"
+version = "2.0.1"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
diff --git a/src/FixedEffectSolvers/FixedEffectSolverGPU.jl b/src/FixedEffectSolvers/FixedEffectSolverGPU.jl
index bff22f7..ab449ce 100644
--- a/src/FixedEffectSolvers/FixedEffectSolverGPU.jl
+++ b/src/FixedEffectSolvers/FixedEffectSolverGPU.jl
@@ -133,11 +133,12 @@ end
 
 function update_weights!(feM::FixedEffectSolverGPU{T}, weights::AbstractWeights) where {T}
 	weights = cu(T, collect(weights))
+	nthreads = feM.m.nthreads
 	for (scale, fe) in zip(feM.m.scales, feM.m.fes)
-		scale!(scale, fe.refs, fe.interaction, weights, fe.m.nthreads)
+		scale!(scale, fe.refs, fe.interaction, weights, nthreads)
 	end
 	for (cache, scale, fe) in zip(feM.m.caches, feM.m.scales, feM.m.fes)
-		cache!(cache, fe.refs, fe.interaction, weights, scale, fe.m.nthreads)
+		cache!(cache, fe.refs, fe.interaction, weights, scale, nthreads)
 	end	
 	feM.weights = weights
 	return feM
@@ -222,4 +223,4 @@ function solve_coefficients!(r::AbstractVector, feM::FixedEffectSolverGPU{T}; to
 	end
 	x = Vector{eltype(r)}[collect(x) for x in feM.x.x]
 	full(normalize!(x, feM.fes; tol = tol, maxiter = maxiter), feM.fes), div(ch.mvps, 2), ch.isconverged
-end
\ No newline at end of file
+end