Login redirect issue with ambari UI and knox sso.

[#1]

I have configured knox with odic and ambari with sso setup as mentioned in articles. While accessing ambari UI with http://knox-host:port/gateway/knoxsso/ambari url I get redirected to oidc provider for authentication, after successful login I again redirected to the ambari login page with the message as " For single sign-on, make sure that Knox Gateway and Ambari Server are located on the same host or subdomain. Alternatively login as an Ambari local user using the local login page." Though my ambari and knox servers on the same machine I get this error. Everything is on aws and I am using public IP address in config.

Please help me to resolve this.

[#2]

I looked into Ambari-server logs and it's saying : JWT authentication failed - Invalid JWT token.