Fix issue when edit address

Repro steps for issue:

1. Create new spree 1.3.1 store, add this extension
2. Sign in, make new order, and checkout
3. Go to My Account page
4. Click 'edit' for your stored address

Throws exception, this fix addresses it.