Adding in test hooks.