Passing correct type now