Fixing another component test