Small fix for SMILE_demo