Small fix to make it properly work.