Fix runs without cuda