Fixes the missing imports