Fix my stupid bug