1 files changed, 2 insertions, 2 deletions
diff --git a/src/main.c b/src/main.c
index 7b3cdf2..d7d30bb 100644
--- a/src/main.c
+++ b/src/main.c
@@ -635,7 +635,7 @@ int set_parameter(
 _Complex double* set_driving(
   nstrophy_parameters parameters
 ){
-  _Complex double* g=calloc(sizeof(_Complex double),(2*parameters.K1+1)*(2*parameters.K2+1));
+  _Complex double* g=calloc(sizeof(_Complex double),parameters.K1*(2*parameters.K2+1)+parameters.K2);
 
   switch(parameters.driving){
   case DRIVING_ZERO:
@@ -664,7 +664,7 @@ _Complex double* set_driving(
 _Complex double* set_init(
   nstrophy_parameters parameters
 ){
-  _Complex double* u0=calloc(sizeof(_Complex double),(2*parameters.K1+1)*(2*parameters.K2+1));
+  _Complex double* u0=calloc(sizeof(_Complex double),parameters.K1*(2*parameters.K2+1)+parameters.K2);
 
   switch(parameters.init){
   case INIT_RANDOM: