Updated code snippet to use with 0.11.4