-# 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
+# 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
+
+if ifclass RELEASE ; then
+ # Remove all FAI logs from chroot via grml-live later then:
+ touch $target/etc/grml_fai_release
+ # Remove all files inside /root/ of chroot:
+ rm -rf $target/root && mkdir -m 0755 $target/root
+fi