From 47dd0df8bc919c9289e9a6434eff5f2e2b4b4452 Mon Sep 17 00:00:00 2001
From: jan <jan@mpxd.net>
Date: Sat, 6 Jan 2018 13:51:42 -0800
Subject: [PATCH] fix operator test

---
 fdfd_tools/eigensolvers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fdfd_tools/eigensolvers.py b/fdfd_tools/eigensolvers.py
index e348cd7..d0ee541 100644
--- a/fdfd_tools/eigensolvers.py
+++ b/fdfd_tools/eigensolvers.py
@@ -53,7 +53,7 @@ def rayleigh_quotient_iteration(operator: sparse.spmatrix or spalg.LinearOperato
     :return: (eigenvalue, eigenvector)
     """
     try:
-        _test = operator - sparse.eye(operator.shape)
+        _test = operator - sparse.eye(operator.shape[0])
         shift = lambda eigval: eigval * sparse.eye(operator.shape[0])
         if solver is None:
             solver = spalg.spsolve