Update unit test spec