Add missing functionality caused by faulty commit.