Fix drawing tests