fixed an edge case