diff --git a/picoc.c b/picoc.c
index cb72cbf..a17f10f 100644
--- a/picoc.c
+++ b/picoc.c
@@ -23,22 +23,27 @@ int main(int argc, char **argv)
     Initialise();
     if (PlatformSetExitPoint())
         return 1;
-        
+    
     PlatformScanFile(argv[1]);
     
     return 0;
 }
 #else
 # ifdef SURVEYOR_HOST
-static char *SourceStr = "\
-printf(\"This is a test program\n\");\
-for (Count = 1; Count <= 10; Count++)\
-    printf(\"%d\n\");\
+static char *SourceStr = "\n\
+int Count;\n\
+\n\
+printf(\"This is a test program\n\");\n\
+for (Count = 1; Count <= 10; Count++)\n\
+    printf(\"%d\n\", Count);\n\
 ";
 
-int main(int argc, char **argv)
+int picoc()
 {    
     Initialise();
+    if (PlatformSetExitPoint())
+        return 1;
+        
     Parse("test.c", SourceStr, strlen(SourceStr), TRUE);
     return 0;
 }
diff --git a/platform.h b/platform.h
index 40c32b7..7d4d1eb 100644
--- a/platform.h
+++ b/platform.h
@@ -2,9 +2,6 @@
 #ifndef PLATFORM_H
 #define PLATFORM_H
 
-/* common includes */
-#include <stdarg.h>
-
 /* configurable options */
 #define UNIX_HOST                           /* select your host type */
 #undef  FLYINGFOX_HOST
@@ -31,6 +28,7 @@
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <unistd.h>
+#include <stdarg.h>
 #ifndef NO_FP
 #include <math.h>
 #endif