Refactored the spec tests to make things a little cleaner

Used a double instead of a custom class