X-Git-Url: https://jasonwoof.com/gitweb/?a=blobdiff_plain;f=su.php;h=2ff23ae01f767d1cfe29ebbfa415df70b8b5a5ab;hb=1656ee459ec82ab8f1113ebc71a8c923517b43ad;hp=0c45cf1d43e42c26e47fe8cb5fc11544c4595737;hpb=b2f3faa3e8e647967e17aac9e9be7ffbad8f9716;p=contractor-progress.git

diff --git a/su.php b/su.php
index 0c45cf1..2ff23ae 100644
--- a/su.php
+++ b/su.php
@@ -18,7 +18,9 @@
 require_once('code/wfpl/format.php');
 
 function su_main() {
-	if(!logged_in_as_contractor()) {
+	if(logged_in_as_contractor()) {
+		set_was_contractor();
+	} elseif(!ever_was_contractor()) {
 		$GLOBALS['url'] = this_url();
 		message('You must be logged in as an administrator to access that function');
 		return 'login';