TODO: Update for 1.0.4