From 197b18f3cffeb4575a188cc526bd6af70d96092c Mon Sep 17 00:00:00 2001
From: Al <albarrentine@gmail.com>
Date: Sat, 9 Jan 2016 01:43:25 -0500
Subject: [PATCH] [fix] NULL check

---
 src/minibatch.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/minibatch.c b/src/minibatch.c
index 2c886876..0a37b72e 100644
--- a/src/minibatch.c
+++ b/src/minibatch.c
@@ -136,11 +136,11 @@ exit_destroy_label_ids:
     free(sorted_keys);
     kh_destroy(str_uint32, label_ids);
     return NULL;
-
 }
 
-
 sparse_matrix_t *feature_matrix(trie_t *feature_ids, feature_count_array *feature_counts) {
+    if (feature_ids == NULL || feature_counts == NULL) return NULL;
+
     const char *feature;
     uint32_t feature_id;
     double count;