Add context to command module tests