Changed .all to .where().load in arrange method