Fix spec so dialyzer passes