diff --git a/simple_admin.theme b/simple_admin.theme
index dcc1111..356276e 100644
--- a/simple_admin.theme
+++ b/simple_admin.theme
@@ -1,6 +1,7 @@
 <?php
 
-/**
+use Drupal\user\Entity\User;
+ /**
  * @file
  * Functions to support theming in the simple_admin theme.
  */
@@ -16,7 +17,7 @@
  if ($uid) {
    $user = \Drupal\user\Entity\User::load($uid);
     // check that the field exists and that it has a value.
-   if (!empty($user->user_picture) && $user->user_picture->isEmpty() === FALSE) {
+ * if (!empty($user->user_picture) && $user->user_picture->isEmpty() === FALSE) {
      $image = $user->user_picture->first();
      $rendered = \Drupal::service('renderer')->renderPlain($image);
    }
@@ -25,7 +26,7 @@
  //dpm($image);
  if(!(\Drupal::currentUser()->isAnonymous())){
   //dpm($variables);
-  $user=\Drupal\user\Entity\User::load(\Drupal::currentUser()->id());
+  $user=User::load(\Drupal::currentUser()->id());
   //dpm($user);
   //$picture=$user->get('user_picture');
   //dpm($picture);