BUG: Add missing imports