Fixing dumb mistake from late night