Add another test to check which module is returned