Fix rake tests