fixed issue with argument types