       .to.contain('menu-open')
 
   it 'navigates forward and back between pages', ->
-    prevControl = @state.div.querySelector('.page-wrapper > .prev')
+    prevControl = @state.div.querySelector('.page-navigation.prev')
     expect(prevControl).to.not.exist # on first page
 
-    nextControl = @state.div.querySelector('.page-wrapper > .next')
+    nextControl = @state.div.querySelector('.page-navigation.next')
     expect(nextControl).to.exist
     expect(nextControl.href).to.contain(SECOND_PAGE)
     page = ReactTestUtils.findRenderedComponentWithType(@state.book, Page)

    expect(_.toArray(@state.div.querySelector('.reference-book').classList))
      .to.contain('menu-open')

    prevControl = @state.div.querySelector('.page-wrapper > .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")
    expect(selection).not.to.be.null
    expect(_.toArray(selection.classList)).to.include('active')


    expect(_.toArray(@state.div.querySelector('.reference-book').classList))
      .to.contain('menu-open')

    nextSelection = @state.div.querySelector(".menu [data-section='#{SECOND_PAGE}'] a")
    ReferenceBookPageActions.loaded(PAGE, SECOND_PAGE_ID)

    commonActions.click(nextSelection)
