Unnecessarily early return to be fixed

I accidentally overlooked this return statement which makes the previous pull request moot. Now it actually updates the context ;)