Fix tests including the correct module (broken after commit 9106d3d)