Fix latest 0.1.10 update (missed 2 lines)