Fix the problem

now it works