Fix test. WIP.