You should change the behaviour as this