Updated for v2.0.6