     expect(_.toArray(@state.div.querySelector('.reference-book').classList))
       .to.contain('menu-open')
 
-    prevControl = @state.div.querySelector('.page-wrapper > .prev')
+    prevControl = @state.div.querySelector('.page-navigation.prev')
     commonActions.click(prevControl, {button: 0})
     expect(_.toArray(@state.div.querySelector('.reference-book').classList))
       .to.contain('menu-open')
 
   it 'sets the menu item to be active based on the current page', ->
-    selection = @state.div.querySelector(".toc [data-section='#{FIRST_PAGE}'] a")
+    selection = @state.div.querySelector(".menu [data-section='#{FIRST_PAGE}'] a")
     expect(selection).not.to.be.null
     expect(_.toArray(selection.classList)).to.include('active')
 

      .to.match(/1/)

  it 'sets the menu item to be active based on the current page', ->
    selection = @state.div.querySelector(".toc [data-section='#{ECO_FIRST_PAGE}'] a")
    expect(selection).not.to.be.null
    expect(_.toArray(selection.classList)).to.include('active')

  it 'navigates forward and back between pages, maintaining ecosystem id', ->
    prevControl = @state.div.querySelector('.page-wrapper > .prev')
    expect(prevControl).to.not.exist # on first page

    nextControl = @state.div.querySelector('.page-wrapper > .next')
    expect(nextControl).to.exist
    expect(nextControl.href).to.contain(ECO_SECOND_PAGE)
    page = ReactTestUtils.findRenderedComponentWithType(@state.book, Page)

    expect(page.context.router.getCurrentParams().section)
      .to.equal(SECOND_PAGE)

    expect(@state.div.querySelector('.page-navigation.next').href)
      .to.contain(THIRD_PAGE)

    expect(@state.div.querySelector('.page-navigation.prev').href)
      .to.contain(FIRST_PAGE)

  it 'preserves menu toggle when navigating forward and back between pages', ->
    toggle = @state.div.querySelector('.menu-toggle')
    nextControl = @state.div.querySelector('.page-navigation.next')

    page = ReactTestUtils.findRenderedComponentWithType(@state.book, Page)
    ReferenceBookPageActions.loaded(PAGE, SECOND_PAGE_ID)
