Fixed test

Fixed the test