Demo stg2 fixes