Updated test for component