Revert "Fix missing import"

This reverts commit 05a677435361ddb74717c0f286a77a8e543744f6.