Code style fix for demo