Fix tests with beaker