diff --git a/src/haxe/ru/m/puzzlez/render/ImagePartBuilder.hx b/src/haxe/ru/m/puzzlez/render/ImagePartBuilder.hx
index 62f6f45..cb69b51 100644
--- a/src/haxe/ru/m/puzzlez/render/ImagePartBuilder.hx
+++ b/src/haxe/ru/m/puzzlez/render/ImagePartBuilder.hx
@@ -22,7 +22,6 @@ class ImagePartBuilder {
     private function buildPart(index:Int, count:Int, parts:Array<Part>, stream:PublicStream<Result>):Void {
         for (i in index...index + count) {
             if (i >= parts.length) {
-                stream.end();
                 return;
             }
             var part = parts[i];
diff --git a/src/haxe/ru/m/puzzlez/render/Render.hx b/src/haxe/ru/m/puzzlez/render/Render.hx
index a5b92aa..d6d31e4 100644
--- a/src/haxe/ru/m/puzzlez/render/Render.hx
+++ b/src/haxe/ru/m/puzzlez/render/Render.hx
@@ -9,6 +9,7 @@ import flash.geom.Rectangle;
 import flash.net.FileReference;
 import flash.utils.ByteArray;
 import haxework.signal.Signal;
+import haxework.view.popup.AlertView;
 import haxework.view.SpriteView;
 import ru.m.puzzlez.core.GameEvent;
 import ru.m.puzzlez.core.GameState;
@@ -68,7 +69,7 @@ class Render extends SpriteView implements IRender {
                 part.complete();
                 imageView.addChild(part);
             case COMPLETE:
-                trace("COMPLETE");
+                AlertView.alert("Complete!");
             case _:
         }
     }
@@ -111,9 +112,10 @@ class Render extends SpriteView implements IRender {
         builder.build(state.parts).then(function(result) {
             parts[result.part.id].image = result.image;
             progress.setProgress(++i, state.parts.length);
-        }).endThen(function(_) {
-            if (progress.content.parent != null) {
-                progress.content.parent.removeChild(progress.content);
+            if (i >= state.parts.length - 1) {
+                if (progress.content.parent != null) {
+                    progress.content.parent.removeChild(progress.content);
+                }
             }
         });
     }