Fix #2

fix!