diff --git a/tools/build/bazel/osgi_java_library.bzl b/tools/build/bazel/osgi_java_library.bzl
index 0352157..7df0ec8 100644
--- a/tools/build/bazel/osgi_java_library.bzl
+++ b/tools/build/bazel/osgi_java_library.bzl
@@ -375,6 +375,10 @@
         include_resources = {},
         test_srcs = None,
         exclude_tests = None,
+        medium_tests = [],
+        large_tests = [],
+        enormous_tests = [],
+        flaky_tests = [],
         test_resources = None,
         visibility = ["//visibility:public"],
         version = ONOS_VERSION,
@@ -494,6 +498,9 @@
             name = name + "-tests-gen",
             test_files = test_srcs,
             exclude_tests = exclude_tests,
+            medium_tests = medium_tests,
+            large_tests = large_tests,
+            enormous_tests = enormous_tests,
             deps = all_test_deps,
         )