Fixed two stale imports