diff --git a/test/gen_random_cone_prob.py b/test/gen_random_cone_prob.py
index 50dbd4a..a22a852 100644
--- a/test/gen_random_cone_prob.py
+++ b/test/gen_random_cone_prob.py
@@ -60,7 +60,7 @@ def pos(x):
 
 
 def get_scs_cone_dims(K):
-    l = K["z"] + K["l"]
+    l = K.get("z", 0) + K.get("l", 0)
     for i in range(0, len(K["q"])):
         l = l + K["q"][i]
 
diff --git a/test/solve_random_cone_prob.py b/test/solve_random_cone_prob.py
index eaa6457..c101dfd 100644
--- a/test/solve_random_cone_prob.py
+++ b/test/solve_random_cone_prob.py
@@ -1,7 +1,6 @@
 from __future__ import print_function, division
 import scs
 import numpy as np
-from scipy import sparse, randn
 import gen_random_cone_prob as tools
 
 #############################################
@@ -28,7 +27,7 @@ def main():
 def solve_feasible(use_indirect, gpu):
     # cone:
     K = {
-        "f": 10,
+        "z": 10,
         "l": 15,
         "q": [5, 10, 0, 1],
         "s": [3, 4, 0, 0, 1],
@@ -50,7 +49,7 @@ def solve_feasible(use_indirect, gpu):
 
 def solve_infeasible(use_indirect, gpu):
     K = {
-        "f": 10,
+        "z": 10,
         "l": 15,
         "q": [5, 10, 0, 1],
         "s": [3, 4, 0, 0, 1],
@@ -66,7 +65,7 @@ def solve_infeasible(use_indirect, gpu):
 
 def solve_unbounded(use_indirect, gpu):
     K = {
-        "f": 10,
+        "z": 10,
         "l": 15,
         "q": [5, 10, 0, 1],
         "s": [3, 4, 0, 0, 1],