Fixed login redirect issue

src/app/pages/login/login.component.ts

@@ -1,5 +1,5 @@
 import { Component, OnInit, NgZone } from '@angular/core';
-import { Auth } from 'aws-amplify';
+import { Auth, Hub } from 'aws-amplify';
 import { Router } from '@angular/router';
 import { AmplifyService } from 'aws-amplify-angular'
 import { NgxSpinnerService } from 'ngx-spinner';
@@ -18,39 +18,36 @@ export class LoginComponent implements OnInit {
     private spinner: NgxSpinnerService) {
 
 
-    // Invoked when congnito call's callback. ie returns to angular page.
-    this.amplifyService.authStateChange$
-      .subscribe(authState => {
-        this.spinner.show();
-
-        if (authState.state === "cognitoHostedUI" || authState.state === "signedIn") {
-          this.zone.run(() => this.router.navigate(['/dashboard']));
-        }else{
-          this.spinner.hide();
-        }
-
-      });
+    // Used for listening to login events
+    Hub.listen("auth", ({ payload: { event, data } }) => {
+      if (event === "cognitoHostedUI" || event === "signedIn") {
+        console.log(event);
+        this.zone.run(() => this.router.navigate(['/dashboard']));
+      } else {
+        this.spinner.hide();
+      }
+    });
 
-      //currentAuthenticatedUser: when user comes to login page again
-      Auth.currentAuthenticatedUser()
-      .then( ()=>{ 
-        this.router.navigate(['/dashboard'],{ replaceUrl: true }); 
-      }).catch( (err) =>{ 
+    //currentAuthenticatedUser: when user comes to login page again
+    Auth.currentAuthenticatedUser()
+      .then(() => {
+        this.router.navigate(['/dashboard'], { replaceUrl: true });
+      }).catch((err) => {
         this.spinner.hide();
         console.log(err);
       })
 
-   }
+  }
 
-  ngOnInit() {}
+  ngOnInit() { }
 
   onLoginClick() {
     this.spinner.show();
     Auth.federatedSignIn();
   }
 
-  
-  onLoginClickOkta(){
+
+  onLoginClickOkta() {
     this.spinner.show();
 
     Auth.federatedSignIn({