Fix demo mode