Spec unit bug fix