From a8ac015bc4e350ffc768699633f757af51a1f03c Mon Sep 17 00:00:00 2001
From: Stephen Birarda <commit@birarda.com>
Date: Tue, 19 Sep 2017 13:56:42 -0700
Subject: [PATCH] address code review comments

---
 assignment-client/src/assets/AssetServer.cpp | 4 ++--
 libraries/image/src/image/Image.cpp          | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/assignment-client/src/assets/AssetServer.cpp b/assignment-client/src/assets/AssetServer.cpp
index 17aae74dfd..fea9d8329d 100644
--- a/assignment-client/src/assets/AssetServer.cpp
+++ b/assignment-client/src/assets/AssetServer.cpp
@@ -280,10 +280,10 @@ void AssetServer::aboutToFinish() {
     // abort each of our still running bake tasks, remove pending bakes that were never put on the thread pool
     auto it = _pendingBakes.begin();
     while (it != _pendingBakes.end()) {
-        auto pendingRunnable =  _bakingTaskPool.tryTake(it.value().get());
+        auto pendingRunnable =  _bakingTaskPool.tryTake(it->get());
 
         if (pendingRunnable) {
-            _pendingBakes.erase(it);
+            it = _pendingBakes.erase(it);
         } else {
             it.value()->abort();
             ++it;
diff --git a/libraries/image/src/image/Image.cpp b/libraries/image/src/image/Image.cpp
index 9d49efcba0..6081355eaa 100644
--- a/libraries/image/src/image/Image.cpp
+++ b/libraries/image/src/image/Image.cpp
@@ -460,6 +460,8 @@ void generateMips(gpu::Texture* texture, QImage& image, const std::atomic<bool>&
             for (int i = 0; i < count; i++) {
                 if (!_abortProcessing.load()) {
                     task(context, i);
+                } else {
+                    break;
                 }
             }
         }