Update tests for token auth