Fix problem with not working