X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?a=blobdiff_plain;f=src%2Fscreen.c;h=927903b13af0bdeee9551be9524373b09e467fd1;hb=976a29bc4205681db1ef50144781d78c4a9b39d0;hp=d55727a4d10d504f01aeec29d5edbdb5216fedfc;hpb=67ecd561eda2d5ca1a9a67588f01bd9d530751c0;p=dosdemo

diff --git a/src/screen.c b/src/screen.c
index d55727a..927903b 100644
--- a/src/screen.c
+++ b/src/screen.c
@@ -8,6 +8,7 @@
 struct screen *tunnel_screen(void);
 struct screen *fract_screen(void);
 struct screen *grise_screen(void);
+struct screen *polytest_screen(void);
 
 #define NUM_SCR	32
 static struct screen *scr[NUM_SCR];
@@ -29,6 +30,9 @@ int scr_init(void)
 	if (!(scr[idx++] = grise_screen())) {
 		return -1;
 	}
+	if(!(scr[idx++] = polytest_screen())) {
+		return -1;
+	}
 	num_screens = idx;
 
 	assert(num_screens <= NUM_SCR);
@@ -113,11 +117,8 @@ int scr_change(struct screen *s, long trans_time)
 		trans_dur = 0;
 	}
 
-	if(cur) {
-		if(cur->stop) {
-			cur->stop(trans_dur);
-		}
-
+	if(cur && cur->stop) {
+		cur->stop(trans_dur);
 		prev = cur;
 		next = s;
 	} else {