Fix shape argument.