Fix ```only``` argument type