Made changes to test