Adds fix for #11 - missing connect