Small fix for router class