Fixed an edge case