Fix that edge case?