1 files changed, 2 insertions, 2 deletions
diff --git a/src/driving.c b/src/driving.c
index 7920d45..527a37d 100644
--- a/src/driving.c
+++ b/src/driving.c
@@ -11,7 +11,7 @@ int g_test(
 ){
   int kx,ky;
   for(kx=0;kx<=K1;kx++){
-    for (ky=-K2;ky<=K2;ky++){
+    for(ky=(kx>0 ? -K2 : 1);ky<=K2;ky++){
       if(kx==2 && ky==-1){
 	g[klookup_sym(kx,ky,K2)]=0.5+sqrt(3)/2*I;
       }
@@ -30,7 +30,7 @@ int g_zero(
   int K2
 ){
   int i;
-  for(i=0;i<(K1+1)*(2*K2+1);i++){
+  for(i=0;i<K1*(2*K2+1)+K2;i++){
     g[i]=0.;
   }