diff --git a/src/sgd.cu b/src/sgd.cu
index d46ece86..f864e6db 100644
--- a/src/sgd.cu
+++ b/src/sgd.cu
@@ -1,3 +1,4 @@
+#include <ctime>
 #include <algorithm>
 #include <vector>
 #include "sgd.h"
@@ -32,7 +33,8 @@ void SGD::Run()
   Tensor xt({(int)maxBatchSize_, (int)numExamples}, 0.0f);
   Tensor yt({(int)maxBatchSize_, (int)numClasses_}, 0.0f);
 
-  vector<size_t> shuffle = CreateShuffle(numExamples);
+  //vector<size_t> shuffle = CreateShuffle(numExamples);
+  vector<size_t> shuffle;
 
   for (size_t numEpoch = 0; numEpoch < epochs_; ++numEpoch) {
     std::cerr << "Starting epoch #" << numEpoch << std::endl;
diff --git a/src/sgd.h b/src/sgd.h
index c5ea8dbc..33364049 100644
--- a/src/sgd.h
+++ b/src/sgd.h
@@ -20,9 +20,9 @@ class SGD {
     void Run();
 
   private:
-    std::shared_ptr<Expr> cost_function_;
-    std::shared_ptr<Expr> inX_;
-    std::shared_ptr<Expr> inY_;
+    Expr *cost_function_;
+    Expr *inX_;
+    Expr *inY_;
     std::vector<Expr*> params_;
     const float eta_;
     std::vector<float>& xData_;