diff --git a/branch-step-05-solution/src/views/PokemonPage.js b/branch-step-05-solution/src/views/PokemonPage.js
index 2970be4..34cbc05 100755
--- a/branch-step-05-solution/src/views/PokemonPage.js
+++ b/branch-step-05-solution/src/views/PokemonPage.js
@@ -1,6 +1,6 @@
 import React from 'react'
 import Relay from 'react-relay'
-import { Link } from 'react-router'
+import { Link, withRouter } from 'react-router'
 import PokemonCard from '../components/PokemonCard'
 import CreatePokemonMutation from '../mutations/CreatePokemonMutation'
 import deleteIcon from '../assets/delete.svg'
@@ -75,7 +75,7 @@ class PokemonPage extends React.Component {
 }
 
 export default Relay.createContainer(
-  PokemonPage,
+  withRouter(PokemonPage),
   {
     initialVariables: {
       id: null,
diff --git a/branch-step-06-solution/src/views/PokemonPage.js b/branch-step-06-solution/src/views/PokemonPage.js
index 0c903df..ec1d7c1 100755
--- a/branch-step-06-solution/src/views/PokemonPage.js
+++ b/branch-step-06-solution/src/views/PokemonPage.js
@@ -1,6 +1,6 @@
 import React from 'react'
 import Relay from 'react-relay'
-import { Link } from 'react-router'
+import { Link, withRouter } from 'react-router'
 import PokemonCard from '../components/PokemonCard'
 import CreatePokemonMutation from '../mutations/CreatePokemonMutation'
 import DeletePokemonMutation from '../mutations/DeletePokemonMutation'
@@ -86,7 +86,7 @@ class PokemonPage extends React.Component {
 }
 
 export default Relay.createContainer(
-  PokemonPage,
+  withRouter(PokemonPage),
   {
     initialVariables: {
       id: null,
diff --git a/branch-step-06/src/views/PokemonPage.js b/branch-step-06/src/views/PokemonPage.js
index 2970be4..34cbc05 100755
--- a/branch-step-06/src/views/PokemonPage.js
+++ b/branch-step-06/src/views/PokemonPage.js
@@ -1,6 +1,6 @@
 import React from 'react'
 import Relay from 'react-relay'
-import { Link } from 'react-router'
+import { Link, withRouter } from 'react-router'
 import PokemonCard from '../components/PokemonCard'
 import CreatePokemonMutation from '../mutations/CreatePokemonMutation'
 import deleteIcon from '../assets/delete.svg'
@@ -75,7 +75,7 @@ class PokemonPage extends React.Component {
 }
 
 export default Relay.createContainer(
-  PokemonPage,
+  withRouter(PokemonPage),
   {
     initialVariables: {
       id: null,
diff --git a/branch-step-07-solution/src/views/PokemonPage.js b/branch-step-07-solution/src/views/PokemonPage.js
index d44cc96..cf478a2 100755
--- a/branch-step-07-solution/src/views/PokemonPage.js
+++ b/branch-step-07-solution/src/views/PokemonPage.js
@@ -1,6 +1,6 @@
 import React from 'react'
 import Relay from 'react-relay'
-import { Link } from 'react-router'
+import { Link, withRouter } from 'react-router'
 import PokemonCard from '../components/PokemonCard'
 import CreatePokemonMutation from '../mutations/CreatePokemonMutation'
 import DeletePokemonMutation from '../mutations/DeletePokemonMutation'
@@ -97,7 +97,7 @@ class PokemonPage extends React.Component {
 }
 
 export default Relay.createContainer(
-  PokemonPage,
+  withRouter(PokemonPage),
   {
     initialVariables: {
       id: null,
diff --git a/branch-step-07/src/views/PokemonPage.js b/branch-step-07/src/views/PokemonPage.js
index 0c903df..ec1d7c1 100755
--- a/branch-step-07/src/views/PokemonPage.js
+++ b/branch-step-07/src/views/PokemonPage.js
@@ -1,6 +1,6 @@
 import React from 'react'
 import Relay from 'react-relay'
-import { Link } from 'react-router'
+import { Link, withRouter } from 'react-router'
 import PokemonCard from '../components/PokemonCard'
 import CreatePokemonMutation from '../mutations/CreatePokemonMutation'
 import DeletePokemonMutation from '../mutations/DeletePokemonMutation'
@@ -86,7 +86,7 @@ class PokemonPage extends React.Component {
 }
 
 export default Relay.createContainer(
-  PokemonPage,
+  withRouter(PokemonPage),
   {
     initialVariables: {
       id: null,