From c06b63211186ee40dfb489f465a2cda4b350337b Mon Sep 17 00:00:00 2001
From: "Veronica K. B. Olsen" <1619840+vkbo@users.noreply.github.com>
Date: Sun, 17 May 2020 16:16:43 +0200
Subject: [PATCH] Import cleanup and fixed test

---
 nw/gui/additions/pageddialog.py | 5 ++---
 tests/test_gui.py               | 4 ++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/nw/gui/additions/pageddialog.py b/nw/gui/additions/pageddialog.py
index 40ba8656..683b62f1 100644
--- a/nw/gui/additions/pageddialog.py
+++ b/nw/gui/additions/pageddialog.py
@@ -29,10 +29,9 @@ import logging
 import nw
 
 from PyQt5.QtCore import Qt, QRect, QPoint
-from PyQt5.QtGui import QPalette
 from PyQt5.QtWidgets import (
-    QWidget, QDialog, QHBoxLayout, QVBoxLayout, QTabWidget, QTabBar,
-    QStylePainter, QStyleOptionTab, QStyle, QLabel
+    QDialog, QHBoxLayout, QVBoxLayout, QTabWidget, QTabBar, QStyle,
+    QStylePainter, QStyleOptionTab
 )
 
 from nw.constants import nwUnicode
diff --git a/tests/test_gui.py b/tests/test_gui.py
index c32e8168..0d73f3c3 100644
--- a/tests/test_gui.py
+++ b/tests/test_gui.py
@@ -276,7 +276,7 @@ def testProjectEditor(qtbot, nwTempGUI, nwRef, nwTemp):
         qtbot.keyClick(projEdit.tabMain.editAuthors, c, delay=keyDelay)
 
     # Test Status Tab
-    projEdit.tabWidget.setCurrentWidget(projEdit.tabStatus)
+    projEdit._tabBox.setCurrentWidget(projEdit.tabStatus)
     projEdit.tabStatus.listBox.item(2).setSelected(True)
     qtbot.mouseClick(projEdit.tabStatus.delButton, Qt.LeftButton)
     qtbot.mouseClick(projEdit.tabStatus.newButton, Qt.LeftButton)
@@ -288,7 +288,7 @@ def testProjectEditor(qtbot, nwTempGUI, nwRef, nwTemp):
     qtbot.mouseClick(projEdit.tabStatus.saveButton, Qt.LeftButton)
 
     # Auto-Replace Tab
-    projEdit.tabWidget.setCurrentWidget(projEdit.tabReplace)
+    projEdit._tabBox.setCurrentWidget(projEdit.tabReplace)
 
     qtbot.mouseClick(projEdit.tabReplace.addButton, Qt.LeftButton)
     projEdit.tabReplace.listBox.topLevelItem(0).setSelected(True)