Tests now run correclty on GPU