bug fix: fixed corner case (maybe numpy bug)