Simplify test by overriding load method