Action modification for tests