Updated test to run properly