Remove test function