Fixed edge cases