Fixing what i broke a few commits ago