BUG: fix validation bug