Fix potential import confusion (in user tests). (#337)

Fix potential import confusion (in user tests). (#337) #47