Fixed from review. Thanks @yoavweiss, @nwtn!