Fix mode.test build. Thanks, Travis!