diff --git a/display.c b/display.c
index 08c9884..c6cb8a3 100644
--- a/display.c
+++ b/display.c
@@ -18,6 +18,7 @@
 #include "efunc.h"
 #include "line.h"
 #include "version.h"
+#include "wrapper.h"
 
 struct video {
 	int v_flag;		/* Flags */
@@ -85,24 +86,13 @@ void vtinit(void)
 	TTopen();		/* open the screen */
 	TTkopen();		/* open the keyboard */
 	TTrev(FALSE);
-	vscreen = (struct video **) malloc(term.t_mrow * sizeof(struct video *));
-
-	if (vscreen == NULL)
-		exit(1);
+	vscreen = xmalloc(term.t_mrow * sizeof(struct video *));
 
 #if	MEMMAP == 0 || SCROLLCODE
-	pscreen = (struct video **) malloc(term.t_mrow * sizeof(struct video *));
-
-	if (pscreen == NULL)
-		exit(1);
+	pscreen = xmalloc(term.t_mrow * sizeof(struct video *));
 #endif
-
 	for (i = 0; i < term.t_mrow; ++i) {
-		vp = (struct video *)malloc(sizeof(struct video) + term.t_mcol);
-
-		if (vp == NULL)
-			exit(1);
-
+		vp = xmalloc(sizeof(struct video) + term.t_mcol);
 		vp->v_flag = 0;
 #if	COLOR
 		vp->v_rfcolor = 7;
@@ -110,11 +100,7 @@ void vtinit(void)
 #endif
 		vscreen[i] = vp;
 #if	MEMMAP == 0 || SCROLLCODE
-		vp = (struct video *)malloc(sizeof(struct video) + term.t_mcol);
-
-		if (vp == NULL)
-			exit(1);
-
+		vp = xmalloc(sizeof(struct video) + term.t_mcol);
 		vp->v_flag = 0;
 		pscreen[i] = vp;
 #endif