Fix spec unit test