Updated API for 0.8.1