bug fixes

2 jobs for tensorflow_group in 3 minutes and 51 seconds (queued for 11 minutes and 53 seconds)