diff --git a/ptypy/custom/LBFGS_cupy.py b/ptypy/custom/LBFGS_cupy.py
index 88a769d34..564a977b3 100644
--- a/ptypy/custom/LBFGS_cupy.py
+++ b/ptypy/custom/LBFGS_cupy.py
@@ -12,10 +12,7 @@
     :license: GPLv2, see LICENSE for details.
 """
 import numpy as np
-# from pycuda import gpuarray
-# import pycuda.driver as cuda
-# import pycuda.cumath
-# from pycuda.tools import DeviceMemoryPool
+import cupy as cp
 
 from ptypy.engines import register
 from ptypy.custom.LBFGS_serial import LBFGS_serial