Fix 5 for GPU forward test