Merge pull request #1 from dmamylin/patch-1

Update polar_histogram.c
agarie · Mar 21, 2015 · d8d0d35 · d8d0d35
2 parents a004023 + ec91ec8
commit d8d0d35
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/polar_histogram.c b/polar_histogram.c
@@ -29,7 +29,10 @@ hist_t * hist_init(int alpha, double threshold, double density_a,
 	hist->densities = (int *)malloc(hist->sectors * sizeof(int));
 
 	/* And is there enough memory for the densities array? */
-	if (NULL == hist->densities) return NULL;
+	if (NULL == hist->densities) {
+		free(hist);
+		return NULL;
+	}
 
 	/* Initialize all densities to 0. */
 	for (int i = 0; i < hist->sectors; ++i) {