Update unit test to match change in species import