Fix awful bug