diff --git a/FunctionSpace/BasisTest.cpp b/FunctionSpace/BasisTest.cpp
index 7ed16815701f6009ba938f1847e0f17f9ea3c842..a76ff1081c13d016fa37727094f51a5c495046d5 100644
--- a/FunctionSpace/BasisTest.cpp
+++ b/FunctionSpace/BasisTest.cpp
@@ -17,7 +17,7 @@
 
 using namespace std;
 
-int main(int argc, char** argv){
+int ain(int argc, char** argv){
   // Init Gmsh //
   GmshInitialize(argc, argv);
 
@@ -28,7 +28,7 @@ int main(int argc, char** argv){
   WriterMsh writer(msh.getGroup(2).getAll());
 
   // Plot Basis //
-  TriEdgeBasis b(5);
+  HexNodeBasis b(1);
   
   PlotBasis plot(msh.getGroup(2), b, writer);
   plot.plot("basis");