diff --git a/lib/obmc_boot_test.py b/lib/obmc_boot_test.py
index 97461c1..073e062 100755
--- a/lib/obmc_boot_test.py
+++ b/lib/obmc_boot_test.py
@@ -10,7 +10,10 @@
 import glob
 import random
 import re
-import cPickle as pickle
+try:
+    import cPickle as pickle
+except ImportError:
+    import pickle
 import socket
 
 from robot.utils import DotDict