Fix reflection issue