diff --git a/tests/ivi_layout-internal-test.c b/tests/ivi_layout-internal-test.c
index 4d73eff1..1054d970 100644
--- a/tests/ivi_layout-internal-test.c
+++ b/tests/ivi_layout-internal-test.c
@@ -33,6 +33,7 @@
 #include <stdint.h>
 
 #include "compositor.h"
+#include "compositor/weston.h"
 #include "ivi-shell/ivi-layout-export.h"
 #include "ivi-shell/ivi-layout-private.h"
 #include "ivi-test.h"
@@ -991,24 +992,18 @@ run_internal_tests(void *data)
 	free(ctx);
 }
 
-int
-controller_module_init(struct weston_compositor *compositor,
-		       int *argc, char *argv[],
-		       const struct ivi_layout_interface *iface,
-		       size_t iface_version);
-
 WL_EXPORT int
-controller_module_init(struct weston_compositor *compositor,
-		       int *argc, char *argv[],
-		       const struct ivi_layout_interface *iface,
-		       size_t iface_version)
+wet_module_init(struct weston_compositor *compositor,
+		       int *argc, char *argv[])
 {
 	struct wl_event_loop *loop;
 	struct test_context *ctx;
+	const struct ivi_layout_interface *iface;
+
+	iface = ivi_layout_get_api(compositor);
 
-	/* strict check, since this is an internal test module */
-	if (iface_version != sizeof(*iface)) {
-		weston_log("fatal: controller interface mismatch\n");
+	if (!iface) {
+		weston_log("fatal: cannot use ivi_layout_interface.\n");
 		return -1;
 	}
 
diff --git a/tests/ivi_layout-test-plugin.c b/tests/ivi_layout-test-plugin.c
index 19eab81a..1f19c55f 100644
--- a/tests/ivi_layout-test-plugin.c
+++ b/tests/ivi_layout-test-plugin.c
@@ -217,25 +217,19 @@ idle_launch_client(void *data)
 	weston_watch_process(&launcher->process);
 }
 
-int
-controller_module_init(struct weston_compositor *compositor,
-		       int *argc, char *argv[],
-		       const struct ivi_layout_interface *iface,
-		       size_t iface_version);
-
 WL_EXPORT int
-controller_module_init(struct weston_compositor *compositor,
-		       int *argc, char *argv[],
-		       const struct ivi_layout_interface *iface,
-		       size_t iface_version)
+wet_module_init(struct weston_compositor *compositor,
+		       int *argc, char *argv[])
 {
 	struct wl_event_loop *loop;
 	struct test_launcher *launcher;
 	const char *path;
+	const struct ivi_layout_interface *iface;
+
+	iface = ivi_layout_get_api(compositor);
 
-	/* strict check, since this is an internal test module */
-	if (iface_version != sizeof(*iface)) {
-		weston_log("fatal: controller interface mismatch\n");
+	if (!iface) {
+		weston_log("fatal: cannot use ivi_layout_interface.\n");
 		return -1;
 	}
 
diff --git a/tests/weston-tests-env b/tests/weston-tests-env
index f62b70ec..f08270e2 100755
--- a/tests/weston-tests-env
+++ b/tests/weston-tests-env
@@ -47,8 +47,7 @@ case $TEST_FILE in
 			--config=$abs_builddir/tests/weston-ivi.ini \
 			--shell=$SHELL_PLUGIN \
 			--socket=test-${TEST_NAME} \
-			--modules=$TEST_PLUGIN \
-			--ivi-module=$MODDIR/${TEST_FILE/.la/.so} \
+			--modules=$TEST_PLUGIN,$MODDIR/${TEST_FILE/.la/.so}\
 			--log="$SERVERLOG" \
 			&> "$OUTLOG"
 		;;