Fix demo saving