Update tests to include nested context instead of describe