diff --git a/views/site.erb b/views/site.erb index b98cd4ef..1fd52e87 100644 --- a/views/site.erb +++ b/views/site.erb @@ -19,13 +19,15 @@ <% if current_site == site %> Edit Site <% end %> - <% if current_site != site %> - <% is_following = current_site.is_following?(site) %> - + + <% if current_site && current_site != site %> + <% is_following = current_site.is_following?(site) %> + <%= is_following ? 'Unfollow' : 'Follow' %> <% end %> +