Tweak tests scheme.

Signed-off-by: Aron Cedercrantz <aron@cedercrantz.se>