Modify connect method call