Fixed missing argument bug

This should work ok now.