Fixing some unit rspecs