diff --git a/models/User.php b/models/User.php
index f8d26d53..95cee70f 100644
--- a/models/User.php
+++ b/models/User.php
@@ -51,8 +51,8 @@ class User extends UserBase
         'street_addr',
         'city',
         'zip',
-        'country',
-        'state'
+        'country_id',
+        'state_id'
     ];
 
     /**