Small improvement (@bgotink)