1 files changed, 29 insertions, 3 deletions
diff --git a/ui/app/select-app.js b/ui/app/select-app.js
index 3cba44052..ffa31b767 100644
--- a/ui/app/select-app.js
+++ b/ui/app/select-app.js
@@ -4,18 +4,44 @@ const connect = require('react-redux').connect
 const h = require('react-hyperscript')
 const App = require('./app')
 const OldApp = require('../../old-ui/app/app')
+const { autoAddToBetaUI } = require('./selectors')
+const { setFeatureFlag } = require('./actions')
 
 function mapStateToProps (state) {
-	return { betaUI: state.metamask.featureFlags.betaUI }
+	return {
+		betaUI: state.metamask.featureFlags.betaUI,
+		autoAdd: autoAddToBetaUI(state),
+		isUnlocked: state.metamask.isUnlocked,
+	}
 }
 
-module.exports = connect(mapStateToProps)(SelectedApp)
+function mapDispatchToProps (dispatch) {
+  return {
+    setFeatureFlagToBeta: () => dispatch(setFeatureFlag('betaUI', true)),
+  }
+}
+module.exports = connect(mapStateToProps, mapDispatchToProps)(SelectedApp)
 
 inherits(SelectedApp, Component)
-function SelectedApp () { Component.call(this) }
+function SelectedApp () {
+	this.state = {
+		autoAdd: false,
+	}
+	Component.call(this)
+}
+
+SelectedApp.prototype.componentWillReceiveProps = function (nextProps) {
+	const { isUnlocked, setFeatureFlagToBeta } = this.props
+
+	if (!isUnlocked && nextProps.isUnlocked && nextProps.autoAdd) {
+		this.setState({ autoAdd: nextProps.autoAdd })
+		setFeatureFlagToBeta()
+	}
+}
 
 SelectedApp.prototype.render = function () {
   const { betaUI } = this.props
+  const { autoAdd } = this.state 
   const Selected = betaUI ? App : OldApp
   return h(Selected)
 }