Fixed regression from last commit.