Fix code types
Fix a sample code