Wrapper fix (final?)