Update shoulda support