3 files changed, 6 insertions, 9 deletions
diff --git a/test/tokeniser.c b/test/tokeniser.c
index 77635bc..913b66f 100644
--- a/test/tokeniser.c
+++ b/test/tokeniser.c
@@ -38,9 +38,8 @@ int main(int argc, char **argv)
 	/* Initialise library */
 	assert(hubbub_initialise(argv[1], myrealloc, NULL) == HUBBUB_OK);
 
-	stream = parserutils_inputstream_create("UTF-8", 0, NULL,
-			myrealloc, NULL);
-	assert(stream != NULL);
+	assert(parserutils_inputstream_create("UTF-8", 0, NULL,
+			myrealloc, NULL, &stream) == PARSERUTILS_OK);
 
 	tok = hubbub_tokeniser_create(stream, myrealloc, NULL);
 	assert(tok != NULL);
diff --git a/test/tokeniser2.c b/test/tokeniser2.c
index 3dee590..d033934 100644
--- a/test/tokeniser2.c
+++ b/test/tokeniser2.c
@@ -141,9 +141,8 @@ void run_test(context *ctx)
 		ctx->output_index = 0;
 		ctx->char_off = 0;
 
-		stream = parserutils_inputstream_create("UTF-8", 0, NULL,
-				myrealloc, NULL);
-		assert(stream != NULL);
+		assert(parserutils_inputstream_create("UTF-8", 0, NULL,
+				myrealloc, NULL, &stream) == PARSERUTILS_OK);
 
 		tok = hubbub_tokeniser_create(stream, myrealloc, NULL);
 		assert(tok != NULL);
diff --git a/test/tokeniser3.c b/test/tokeniser3.c
index 8febda9..b7ca15e 100644
--- a/test/tokeniser3.c
+++ b/test/tokeniser3.c
@@ -138,9 +138,8 @@ void run_test(context *ctx)
 		ctx->output_index = 0;
 		ctx->char_off = 0;
 
-		stream = parserutils_inputstream_create("UTF-8", 0, NULL,
-				myrealloc, NULL);
-		assert(stream != NULL);
+		assert(parserutils_inputstream_create("UTF-8", 0, NULL,
+				myrealloc, NULL, &stream) == PARSERUTILS_OK);
 
 		tok = hubbub_tokeniser_create(stream, myrealloc, NULL);
 		assert(tok != NULL);