Add error behaviour