Fix imports not being used