fixing non-cuda build