Bug fix: forgot to pass argument  to .