Updated test

Updated test