Set the federation type to SAML.
PS> Set-MsolDomainAuthentication -DomainName “$domainName” 
            -Authentication Federated -PreferredAuthenticationProtocol Samlp