Index: rooms_booking.availability_agent.inc
===================================================================
--- rooms_booking.availability_agent.inc	(revision 3848)
+++ rooms_booking.availability_agent.inc	(working copy)
@@ -67,7 +67,8 @@
    * each set.
    */
   private function searchForAvailability($rooms_results) {
-    $el_key = array_shift(array_keys($rooms_results));
+  	$allkeys = array_keys($rooms_results);
+    $el_key = array_shift($allkeys);
 
     if (!isset($rooms_results[$el_key]['rooms_unit'])) return 0;
 
@@ -182,8 +183,9 @@
 
     // Load the unit
     $unit = rooms_unit_load($unit_id);
-
-    $units = array_pop(array_pop($this->getUnitsByPriceType(array('rooms_unit' => array($unit_id => $unit)))));
+		$units_by_price = $this->getUnitsByPriceType(array('rooms_unit' => array($unit_id => $unit)));
+		$unit_by_price = array_pop($units_by_price);
+    $units = array_pop($unit_by_price);
     
     if (count($units) == 0) {
       return ROOMS_NO_ROOMS;