Adds necessary import removed by previous commit.