Fix code to work with 0.1.9