Attempt to fix location entry test

It's no longer a GtkEntry

1 files changed, 3 insertions, 2 deletions

diff --git a/tests/ephy-location-entry-test.c b/tests/ephy-location-entry-test.c
index bcd6bebaa..c378187a5 100644
--- a/tests/ephy-location-entry-test.c
+++ b/tests/ephy-location-entry-test.c
@@ -39,9 +39,10 @@ static void
 test_entry_get_entry (void)
 {
   EphyLocationEntry *entry;
+
   entry = EPHY_LOCATION_ENTRY (ephy_location_entry_new ());
 
-  g_assert (GTK_IS_ENTRY (entry));
+  g_assert (GTK_IS_ENTRY (ephy_location_entry_get_entry (entry)));
 }
 
 static void
@@ -111,7 +112,7 @@ test_entry_can_undo (void)
   /* Use gtk_* function or otherwise user_changed won't be correctly handled
    * internally by the location entry (see editable_changed_cb and
    * block_update) */
-  gtk_entry_set_text (GTK_ENTRY (entry), test);
+  gtk_entry_set_text (GTK_ENTRY (ephy_location_entry_get_entry (entry)), test);
   g_assert_cmpint (ephy_location_entry_get_can_undo (entry), ==, TRUE);
 }