Handle corner case where flatten function returns compound output.

Handle corner case where flatten function returns compound output. #87