Fix #124 tests context code 

Added the `code` object to each `context`