BUG: fixed bug, thanks @jairideout