GPU test update

Needs to be tested on Linux