Fix imports to work as package