diff --git a/src/flowtype/index.js b/src/flowtype/index.js
index 88445c32..be8638cf 100644
--- a/src/flowtype/index.js
+++ b/src/flowtype/index.js
@@ -119,15 +119,8 @@ type UiEventAction = {
     // },
 };
 
-// TODO: join this message with uiMessage
-type IFrameHandshake = {
-    type: 'iframe_handshake',
-    payload: any,
-};
-
 export type Action =
     | RouterAction
-    | IFrameHandshake
     | TransportEventAction
     | DeviceEventAction
     | UiEventAction
diff --git a/src/reducers/TrezorConnectReducer.js b/src/reducers/TrezorConnectReducer.js
index 7d087f18..135a1dd4 100644
--- a/src/reducers/TrezorConnectReducer.js
+++ b/src/reducers/TrezorConnectReducer.js
@@ -1,5 +1,5 @@
 /* @flow */
-import { TRANSPORT, UI } from 'trezor-connect';
+import { TRANSPORT, IFRAME } from 'trezor-connect';
 import * as CONNECT from 'actions/constants/TrezorConnect';
 
 import type { Action } from 'flowtype';
@@ -66,7 +66,7 @@ export default function connect(state: State = initialState, action: Action): St
                 error: action.error,
             };
         // trezor-connect iframe loaded
-        case UI.IFRAME_HANDSHAKE:
+        case IFRAME.LOADED:
             return {
                 ...state,
                 initialized: true,