Fix import errors when installed from source