Fixes nil context errors