Fix unit tests not running.

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