Fixed unit test timing error