Run setup_test_links in both deps

TEST=Ran it
BUG=None

Change-Id: I8aca7639ca3461c3f108e6faf60111ce4ed5b4af
Reviewed-on: https://gerrit.chromium.org/gerrit/26631
Tested-by: Kris Rambish <krisr@chromium.org>
Reviewed-by: Nirnimesh <nirnimesh@chromium.org>
diff --git a/mod_test_image_for_pyauto.sh b/mod_test_image_for_pyauto.sh
index c7e5d18..6afc26a 100644
--- a/mod_test_image_for_pyauto.sh
+++ b/mod_test_image_for_pyauto.sh
@@ -162,6 +162,9 @@
   sudo chown root:root suid-python
   sudo chmod 4755 suid-python
   popd
+  info "Setting up pyauto required symbolic links in $item..."
+  sudo chroot "${ROOT_FS_DIR}" sudo bash "/usr/local/autotest/deps/\
+$item/test_src/out/Release/setup_test_links.sh"
 done
 
 # Add an easy link to get to the functional folder
@@ -169,11 +172,6 @@
   "/usr/local/autotest/deps/chrome_test/test_src/chrome/test/functional" \
   "${ROOT_FS_DIR}/pyauto"
 
-info "Setting up pyauto required symbolic links..."
-SETUP_LINKS="/usr/local/autotest/deps/chrome_test/test_src/out/\
-Release/setup_test_links.sh"
-sudo chroot "${ROOT_FS_DIR}" sudo bash "${SETUP_LINKS}"
-
 cleanup
 
 # cros_make_image_bootable is unstable (crosbug.com/18709)