Clean up behat method