A small hack for my problem.