Fixed type import confusion