Made the spec test pass :)