Rewrote tests to use equal instead of ok