Its actually done in same function but I can change it to separate function since it is another test.