Move stub in the test class