diff --git a/docs/SnpArrays.ipynb b/docs/SnpArrays.ipynb
index 3fb7ba96..a4685f0c 100644
--- a/docs/SnpArrays.ipynb
+++ b/docs/SnpArrays.ipynb
@@ -62,6 +62,7 @@
    "outputs": [],
    "source": [
     "# for use in this tutorial\n",
+
     "using SnpArrays, ADMIXTURE, BenchmarkTools, DelimitedFiles, Glob, Random\n",
     "Sys.islinux() && (using CUDA);"
    ]
diff --git a/src/linalg_direct.jl b/src/linalg_direct.jl
index ae489b2f..e3388794 100644
--- a/src/linalg_direct.jl
+++ b/src/linalg_direct.jl
@@ -60,7 +60,7 @@ eltype(bm::SnpLinAlg) = eltype(bm.μ)
 
 function Base.getindex(s::SnpLinAlg{T}, i::Int, j::Int) where T
     x = SnpArrays.convert(T, getindex(s.s, i, j), s.model)
-    s.impute && isnan(x) && return s.μ[j]
+    s.impute && isnan(x) && (x = s.μ[j])
     s.center && (x -= s.μ[j])
     s.scale && (x *= s.σinv[j])
     return x