diff --git a/mmdet/models/detectors/base.py b/mmdet/models/detectors/base.py
index e617b0e306df9364ce0b736c03acb0136b4c2056..6d26dc3a5aba773e521f3ffdcaf9ee7958b88843 100644
--- a/mmdet/models/detectors/base.py
+++ b/mmdet/models/detectors/base.py
@@ -3,7 +3,6 @@ from abc import ABCMeta, abstractmethod
 
 import mmcv
 import numpy as np
-import torch
 import torch.nn as nn
 
 from mmdet.core import tensor2imgs, get_classes
diff --git a/tools/test.py b/tools/test.py
index e1552e58209458b65e69c740619d789b1b8e4cb2..b322bb203f401775565065768adbd8acab57eba4 100644
--- a/tools/test.py
+++ b/tools/test.py
@@ -32,7 +32,7 @@ def single_test(model, data_loader, show=False):
 
 def _data_func(data, device_id):
     data = scatter(collate([data], samples_per_gpu=1), [device_id])[0]
-    return dict(**data, return_loss=False, rescale=True)
+    return dict(return_loss=False, rescale=True, **data)
 
 
 def parse_args():