diff --git a/source/module_psi/psi.cpp b/source/module_psi/psi.cpp
index 77df3a6263..26dbdb09a1 100644
--- a/source/module_psi/psi.cpp
+++ b/source/module_psi/psi.cpp
@@ -281,6 +281,7 @@ template <typename T, typename Device> int Psi<T, Device>::get_current_nbas() co
 
 template <typename T, typename Device> const int& Psi<T, Device>::get_ngk(const int ik_in) const
 {
+    if (!this->ngk) return this->nbasis;
     return this->ngk[ik_in];
 }