small fix to make it work