@@ -375,12 +375,17 @@ class WifiConnectPage(UbuntuUIToolkitCustomProxyObjectBase):
375375 wizard = get_wizard(self)
376376 next_page = wizard.get_current_page()
377377 locationPageEnabled = True
378+ reportingPageEnabled = True
378379 if next_page.objectName == 'locationPage':
379380 next_page = wizard.get_location_page()
380381 else:
381382 locationPageEnabled = False
382− next_page = wizard.get_reporting_page()
383− return locationPageEnabled, next_page
383+ if next_page.objectName == 'reportingPage':
384+ next_page = wizard.get_reporting_page()
385+ else:
386+ reportingPageEnabled = False
387+ next_page = wizard.get_finished_page()
388+ return locationPageEnabled, reportingPageEnabled, next_page
384389
385390 def _get_notification(self, unity):
386391 logger.info('Waiting longer for notification object')