small Update to #12 to fix fatal error