Fixes missing import and arg parser reference