Fixing ancient mistake (forgot I'm using old code)