Fix compile issue in sample