fixing to work with 0.6