minor fix
Merge #79. Fix #49