Fix @todb-r7 feedback