Fix tests (shame on me)