Converting LensTest to be type safe
Revised type signature of StateM.put