diff --git a/test/tests/Test_CodeGeneration/NamespacedFunctions/test.cpp b/test/tests/Test_CodeGeneration/NamespacedFunctions/test.cpp
index 785819c..1eaec05 100644
--- a/test/tests/Test_CodeGeneration/NamespacedFunctions/test.cpp
+++ b/test/tests/Test_CodeGeneration/NamespacedFunctions/test.cpp
@@ -10,14 +10,16 @@ class Class {
   ~Class() {}
 };
 
-Class getClass(const int val) {
+// TODO: Return the `Class` object.
+int getClass(const int val) {
   Class output(val);
-  return output;
+  return output.m_val;
 }
 
 } // namespace Namespace
 
-Namespace::Class getClass(const int val) { return Namespace::getClass(val); }
+// TODO: Use this function.
+int getClass(const int val) { return Namespace::getClass(val); }
 
 int main(int argc, char **argv) {
   Namespace::Class obj = Namespace::getClass(5);