1 files changed, 4 insertions, 1 deletions
diff --git a/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java b/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java
index e6319c9..12b5701 100644
--- a/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java
+++ b/src/main/java/org/openslx/imagemaster/serverconnection/ConnectionHandler.java
@@ -186,8 +186,10 @@ public class ConnectionHandler implements IncomingEvent
 							}
 							// get start of range.
 							int blockNumber = image.getNextMissingBlock();
-							if ( blockNumber == -1 )
+							if ( blockNumber == -1 ) {
+								log.debug( "Download complete." );
 								return null;
+							}
 							lastBlock.setValue( blockNumber );
 							log.debug( "Block " + blockNumber + " was transmitted " + image.getTimesTransmitted( blockNumber ) + " time(s)." );
 
@@ -197,6 +199,7 @@ public class ConnectionHandler implements IncomingEvent
 							return range;
 						}
 					} );
+					image.updateDb();
 				}
 			} );
 		} catch ( RejectedExecutionException e ) {