Modificar functions.php de Wordpress para controlar Acceso de Usuarios

Tengo un código en [login to view URL] de Wordpress capaz de controlar el acceso a la web mediante un token. Una vez que se accede a la web con dicho token, hay un pequeño error, y es que en los enlaces en token no se mantiene, por lo que se me sale de la web, ya que para navegar a esta solo se puede hacer mediante dicho token. Esto sería lo que habría que corregir. Dicho problema, para que una vez estemos dentro, podamos navegar por los diferentes enlaces sin salirse. Tengo una web de prueba instalada para poder ver el funcionamiento de dicho token y realizar pruebas sobre esta si asi lo requieres. Muchas gracias