fix size check in plot brain net