Use respond_to block
Fix test