Fixed unit test to run