Fixed tflatMap impl