Fix test for specifying port