Fix issue with int args in test