Update Base.java

Making tests more generic.