Fixed failing RPC flattening test for new axis operation