Fix handling of edge case