Fix milestone bug