Update to fix #205