Fix initialiation with GPU ID