-# final cleanup
-cp /etc/grml/fai/grml/grml_cleanup_chroot $target/root/
-$ROOTCMD sh -c "NOCOLORS=1 /root/grml_cleanup_chroot"
-rm $target/root/grml_cleanup_chroot
+# misc cleanup:
+if [ -f /etc/grml/fai/grml/grml_cleanup_chroot ] ; then
+ cp /etc/grml/fai/grml/grml_cleanup_chroot $target/root/
+ $ROOTCMD /root/grml_cleanup_chroot
+ rm $target/root/grml_cleanup_chroot
+fi
+
+# make sure GRML_SMALL uses the appropriate configuration:
+if ifclass GRML_SMALL ; then
+ cp $target/etc/inittab.small $target/etc/inittab
+ cp $target/etc/runlevel.conf.livecd.small $target/etc/runlevel.conf
+fi