Improve puma feature check test.