Please create an `ensure_connection()` method or something, this is raelly ugly.