Restore testing ability

Oops