Add explicit key module load to test module