diff --git invite.module invite.module
index 0a926ee..bf44076 100644
--- invite.module
+++ invite.module
@@ -314,8 +314,8 @@ function invite_user_access($permission, $account) {
  *   The user id to check accepted invitations for.
  */
 function invite_notify($uid) {
-  $result = db_query('SELECT invitee FROM {invite_notifications} WHERE uid = :uid', array(':uid' => $uid));
-  while ($row = db_fetch_object($result)) {
+  $result = db_query('SELECT invitee FROM {invite_notifications} WHERE uid = :uid', array(':uid' => $uid))->fetchAll();
+  foreach($result as $row) {
     $account = user_load_multiple(array($row->invitee), array('status' => 1));
     if ($account) {
       drupal_set_message(t('!user (@email) has joined @site-name!', array('!user' => theme('username', array('account' => $account)), '@email' => $account->mail, '@site-name' => variable_get('site_name', t('Drupal')))));
@@ -413,8 +413,7 @@ function invite_form_alter(&$form, &$form_state, $form_id) {
  *   An invite record.
  */
 function invite_load($code) {
-  $result = db_query("SELECT * FROM {invite} WHERE reg_code = :reg_code AND canceled = 0", array(':reg_code' => $code));
-  if ($invite = db_fetch_object($result)) {
+  if ($invite = db_query("SELECT * FROM {invite} WHERE reg_code = :reg_code AND canceled = 0", array(':reg_code' => $code))->fetch()) {
     $invite->inviter = user_load($invite->uid);
     $invite->data = (array) unserialize($invite->data);
   }