Forgot to put logic back for testing.