Make the sample code actually work