Add bake tests