Fix overridden test

Fix overridden test