Oops fix edge case