Upgrade tests to 1.0.4