Fix ugly error