Fixing what I messed up