Update tests to use Puma