Fixed some breaking specs