Retain the reference color space (TextureGroup#784)

AtomTickets · Apr 25, 2018 · ef1d6f6 · ef1d6f6
1 parent 0063b2a
commit ef1d6f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Source/Details/ASGraphicsContext.m b/Source/Details/ASGraphicsContext.m
@@ -154,7 +154,8 @@ extern void ASGraphicsBeginImageContextWithOptions(CGSize size, BOOL opaque, CGF
   dispatch_once(&onceToken, ^{
     UIGraphicsBeginImageContextWithOptions(CGSizeMake(1, 1), YES, 0);
     UIImage *refImage = UIGraphicsGetImageFromCurrentImageContext();
-    imageColorSpace = CGImageGetColorSpace(refImage.CGImage);
+    imageColorSpace = CGColorSpaceRetain(CGImageGetColorSpace(refImage.CGImage));
+    ASDisplayNodeCAssertNotNil(imageColorSpace, nil);
     UIGraphicsEndImageContext();
   });