Refactoring project

Fixing an issue that prevented me from properly compiling the code,
therefore being unable to test the project.