Fix test to take a list of layers.