Fix missing `return self` in __enter__
Return the internal res.allocation id, so it can be modified through other allocation objects