Fix spec after changing. Might wanna run rspec next time