Fix mem leak (0.1.6)

Fixed a memory leak