Fixed arguments issue
Update for arg fix