Fix two gpu sanity tests