Beter context handling