Replace Contexts with the one provided by the test package. Signed-off-by: Ivan Valdes <ivan@vald.es>