Update test
remove cast
update test