Adds yet another forgotten import