quick fix to make it work.