fixed annoying numpy dimension bug