From 7aca45c46fa50756331156ca51ee793b9240a465 Mon Sep 17 00:00:00 2001
From: Ives van Hoorne <ives.v.h@gmail.com>
Date: Thu, 13 Dec 2018 18:01:14 +0100
Subject: [PATCH] Fix preview SSE screenshot not hiding

---
 packages/app/src/app/components/Preview/elements.js | 2 ++
 packages/app/src/app/components/Preview/index.js    | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/packages/app/src/app/components/Preview/elements.js b/packages/app/src/app/components/Preview/elements.js
index 6a747d72f91..362a39e08b4 100644
--- a/packages/app/src/app/components/Preview/elements.js
+++ b/packages/app/src/app/components/Preview/elements.js
@@ -41,4 +41,6 @@ export const Loading = styled.div`
   color: white;
   line-height: 1.3;
   text-align: center;
+
+  z-index: 10;
 `;
diff --git a/packages/app/src/app/components/Preview/index.js b/packages/app/src/app/components/Preview/index.js
index 56ce63ea2c6..695140b29ca 100644
--- a/packages/app/src/app/components/Preview/index.js
+++ b/packages/app/src/app/components/Preview/index.js
@@ -191,6 +191,11 @@ class BasePreview extends React.Component<Props, State> {
       this.connectTimeout = null;
       this.localClose = false;
       this.setupSSESockets();
+
+      setTimeout(() => {
+        // Remove screenshot after specific time, so the loading container spinner can still show
+        this.setState({ showScreenshot: false });
+      }, 100);
     }
     this.listener = listen(this.handleMessage);