extend feature to ensure it works with parameters as well