I think I'd make sure he's eating well and has his strength back before putting him back in deeper water.
(You're changing his little container's water daily, right ?)

~Mac