Make testing functiono work