Add tests for unsubscribing