Fixed something I had messed up last commit