Fix #2

Maybe this