Fixing dumb bug