fix for GPU test