Forgot to map func for flat network