fix test logic :grin: