spec fix, little logic