Fix from last commit :-(