typo/comments
Fix tests due to change to gradient check dtype.