Fix reported output dimensions in certain cases
v0.5.1-pl