fix(flow): Move import back to functions