Fixes

• Fixing issue with flow where a type was being used without being imported #677