Which module to use with impuls switches

I have 3 light circuits which were powered on by using a button press switch and now I want to use these buttons to activate a zigbee module which stays on for a time and which can be turned off again after pressing the button again.
If I now press the button it can activate a zigbee switch but when releasing the button also the zigbee module goes off and I want it to stay on until the next press of the impuls button.