I think I did it. Need some workaround