Reverted the removal of the test dependency