diff --git a/faiss/gpu/test/torch_test_contrib_gpu.py b/faiss/gpu/test/torch_test_contrib_gpu.py
index f7444337f1..f1a92c33b3 100644
--- a/faiss/gpu/test/torch_test_contrib_gpu.py
+++ b/faiss/gpu/test/torch_test_contrib_gpu.py
@@ -3,10 +3,10 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
-import faiss
 import torch
 import unittest
 import numpy as np
+import faiss
 import faiss.contrib.torch_utils
 
 def to_column_major_torch(x):
diff --git a/faiss/python/swigfaiss.swig b/faiss/python/swigfaiss.swig
index 43c8d6c422..752b64b55b 100644
--- a/faiss/python/swigfaiss.swig
+++ b/faiss/python/swigfaiss.swig
@@ -352,6 +352,8 @@ void gpu_sync_all_devices()
 %include  <faiss/gpu-rocm/GpuResources.h>
 %include  <faiss/gpu-rocm/StandardGpuResources.h>
 
+typedef ihipStream_t* hipStream_t;
+
 %inline %{
 
 // interop between pytorch exposed hipStream_t and faiss