1 files changed, 6 insertions, 2 deletions
diff --git a/test/functional/ui/screen.lua b/test/functional/ui/screen.lua
index c60944bbb0..174538df6e 100644
--- a/test/functional/ui/screen.lua
+++ b/test/functional/ui/screen.lua
@@ -190,8 +190,11 @@ function Screen:set_default_attr_ignore(attr_ignore)
   self._default_attr_ignore = attr_ignore
 end
 
-function Screen:attach()
-  request('ui_attach', self._width, self._height, true)
+function Screen:attach(rgb)
+  if rgb == nil then
+    rgb = true
+  end
+  request('ui_attach', self._width, self._height, rgb)
 end
 
 function Screen:detach()
@@ -500,6 +503,7 @@ function Screen:snapshot_util(attrs, ignore)
   else
     print( "]], "..attrstr..")\n")
   end
+  io.stdout:flush()
 end
 
 function pprint_attrs(attrs)