fix to edge case